docs

kevinsbello · Sep 1, 2023 · 2e39dd3 · 2e39dd3
1 parent e1395b6
commit 2e39dd3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/docs/index.rst b/docs/index.rst
@@ -108,6 +108,7 @@ and we can observe how the solution at the final iteration (bottom-right plot) i
    :caption: Getting Started
    :hidden:
 
+   Home<self>
    usage
 
 .. toctree::

diff --git a/src/dagma/linear.py b/src/dagma/linear.py
@@ -355,7 +355,7 @@ def fit(self,
         return self.W_est
 
 def test():
-    import utils
+    from . import utils
     from timeit import default_timer as timer
     utils.set_random_seed(1)
 

diff --git a/src/dagma/nonlinear.py b/src/dagma/nonlinear.py
@@ -333,7 +333,7 @@ def fit(self,
 
 def test():
     from timeit import default_timer as timer
-    import utils
+    from . import utils
 
     utils.set_random_seed(1)
     torch.manual_seed(1)