diff --git a/docs/api.rst b/docs/api.rst index 653d1876b..61bb92117 100644 --- a/docs/api.rst +++ b/docs/api.rst @@ -20,4 +20,5 @@ API Reference simba.unsupervised simba.utils simba.utils.cli - simba.video_processors \ No newline at end of file + simba.video_processors + simba.gpu_helpers \ No newline at end of file diff --git a/docs/simba.rst b/docs/simba.rst index 5a0408c46..ae9bf220e 100644 --- a/docs/simba.rst +++ b/docs/simba.rst @@ -20,3 +20,4 @@ SimBA modules simba.unsupervised simba.utils simba.video_processors + simba.gpu_helpers