Update to aind-data-schema==1.1.1

AllenInstitute · Oct 26, 2024 · a09ac6f · a09ac6f
1 parent 283e999
commit a09ac6f
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 12 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,6 +24,7 @@ dependencies = [
     "hdmf>=3.14.0",
     "ndx-pose>=0.1.1",
     "tables>=3.9.2",
+    "aind-data-schema==1.1.1",
     "pydantic>=2.6.4",
     "npc-lims>=0.1.183",
     "npc-sync>=0.1.19",
@@ -33,7 +34,6 @@ dependencies = [
     "npc-session>=0.1.39",
     "npc-mvr>=0.1.8",
     "npc-io>=0.1.30",
-    "aind-data-schema @ git+https://github.com/AllenNeuralDynamics/aind-data-schema.git@1072-qc-automate-state-properties-in-qualitycontrol-and-evaluation-based-on-metric-states",
 ]
 requires-python = ">=3.9"
 readme = "README.md"

diff --git a/src/npc_sessions/sessions.py b/src/npc_sessions/sessions.py
@@ -3357,7 +3357,7 @@ def _aind_data_streams(self) -> tuple[aind_data_schema.core.session.Stream, ...]
                     ),
                     ephys_modules=(
                         ephys_modules := [
-                            aind_data_schema.core.session.EphysModule(
+                            aind_data_schema.core.session.ManipulatorModule(
                                 assembly_name=probe.name.upper(),
                                 arc_angle=0.0,
                                 module_angle=0.0,
@@ -3384,11 +3384,6 @@ def _aind_data_streams(self) -> tuple[aind_data_schema.core.session.Stream, ...]
                                         unit="micrometer",
                                     )
                                 ),
-                                ephys_probes=[
-                                    aind_data_schema.core.session.EphysProbeConfig(
-                                        name=probe.name.upper(),
-                                    )
-                                ],
                             )
                             for probe in self.probe_letters_to_use
                         ]