refactor: remove temp dataset

Author: arjunsridhar12345

pyproject.toml

@@ -52,7 +52,7 @@ extend-ignore = []
 convention = 'google'
 
 [tool.coverage.run]
-omit = ["*__init__*", "*/temp_dataset.py"]
+omit = ["*__init__*"]
 source = ["dynamic_foraging_processing", "tests"]
 
 [tool.coverage.report]

src/dynamic_foraging_processing/raw_data_loader/loader.py

@@ -5,8 +5,7 @@
 import typing as t
 
 import pandas as pd
-
-from .temp_dataset import dataset as _build_dataset
+from aind_behavior_dynamic_foraging.data_contract import dataset as _build_dataset
 
 if t.TYPE_CHECKING:
     from contraqctor.contract import Dataset