#291 Fix imports in tests/e2e/helpers.py

PainterQubits · Jun 26, 2024 · e1a7152 · e1a7152
1 parent e4e36ef
commit e1a7152
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/e2e/helpers.py b/tests/e2e/helpers.py
@@ -8,7 +8,7 @@
 from sqlalchemy import delete
 from freezegun import freeze_time
 import astropy.units as u  # type: ignore
-from paramdb import ParamDB, Param, Struct, ParamList, ParamDict
+from paramdb import ParamDB, ParamDataclass, ParamList, ParamDict
 from paramdb._database import _Snapshot
 
 
@@ -18,14 +18,14 @@
 _DB = ParamDB[Any](DB_PATH)
 
 
-class CustomParam(Param):
+class CustomParam(ParamDataclass):
     """Custom parameter."""
 
     int: int
     str: str
 
 
-class CustomStruct(Struct):
+class CustomStruct(ParamDataclass):
     """Custom parameter structure."""
 
     int: int