fix database import

neuromodulation · Sep 19, 2024 · 37ea063 · 37ea063
1 parent 96240c3
commit 37ea063
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/py_neuromodulation/stream/stream.py b/py_neuromodulation/stream/stream.py
@@ -227,7 +227,7 @@ def run(
 
         # Open database connection
         # TONI: we should give the user control over the save format
-        from py_neuromodulation.nm_database import NMDatabase
+        from py_neuromodulation.utils.database import NMDatabase
 
         db = NMDatabase(experiment_name, out_dir)  # Create output database
 

diff --git a/tests/test_database.py b/tests/test_database.py
@@ -1,9 +1,9 @@
-from py_neuromodulation import nm_database
+from py_neuromodulation.utils import database
 import pandas as pd
 
 
 def test_db_setup():
-    nm_db = nm_database.NMDatabase(
+    nm_db = database.NMDatabase(
         name="test_db",
         out_dir="test_data",
     )