From e1a715221bdc5a65662db0f18670251e83cf4862 Mon Sep 17 00:00:00 2001
From: Alex Hadley <alexhad6@users.noreply.github.com>
Date: Wed, 26 Jun 2024 12:45:38 -0700
Subject: [PATCH] #291 Fix imports in tests/e2e/helpers.py

---
 tests/e2e/helpers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/e2e/helpers.py b/tests/e2e/helpers.py
index db649c4..7246db8 100644
--- 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