Linting

docs/conf.py

@@ -80,4 +80,4 @@ def export_model_diagram(model: BaseModel, root: str = "_static") -> None:
 
 
 _diagram_root = "_static"
-#export_model_diagram(MyPydanticBaseModel, _diagram_root)
+# export_model_diagram(MyPydanticBaseModel, _diagram_root)

examples/example.py

@@ -2,9 +2,12 @@
 import os
 
 from aind_behavior_services.session import AindBehaviorSessionModel
-from aind_behavior_device_water_tuner.task_logic import AindBehaviorWaterTunerParameters, AindBehaviorWaterTunerTaskLogic
-import aind_behavior_device_water_tuner.rig import AindBehaviorWaterTunerRig
 
+from aind_behavior_device_water_tuner.rig import AindBehaviorWaterTunerRig
+from aind_behavior_device_water_tuner.task_logic import (
+    AindBehaviorWaterTunerParameters,
+    AindBehaviorWaterTunerTaskLogic,
+)
 
 
 def mock_session() -> AindBehaviorSessionModel:
@@ -23,17 +26,13 @@ def mock_session() -> AindBehaviorSessionModel:
 
 
 def mock_rig() -> AindBehaviorWaterTunerRig:
-
     return AindBehaviorWaterTunerRig(
         rig_name="test_rig",
     )
 
 
 def mock_task_logic() -> AindBehaviorWaterTunerTaskLogic:
-
-    return AindBehaviorWaterTunerTaskLogic(
-        task_parameters=AindBehaviorWaterTunerParameters()
-    )
+    return AindBehaviorWaterTunerTaskLogic(task_parameters=AindBehaviorWaterTunerParameters())
 
 
 def main(path_seed: str = "./local/{schema}.json"):
@@ -49,4 +48,4 @@ def main(path_seed: str = "./local/{schema}.json"):
 
 
 if __name__ == "__main__":
-    main()
+    main()

src/aind_behavior_device_water_tuner/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.0"
+__version__ = "0.0.0"

src/aind_behavior_device_water_tuner/launcher.py

@@ -36,4 +36,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

src/aind_behavior_device_water_tuner/regenerate.py

@@ -35,4 +35,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

src/aind_behavior_device_water_tuner/rig.py

@@ -1,11 +1,10 @@
 from typing import Literal
 
-from aind_behavior_services.rig import AindBehaviorRigModel
-from aind_behavior_services.calibration import aind_manipulator
 import aind_behavior_services.rig as rig
+from aind_behavior_services.calibration import aind_manipulator
+from aind_behavior_services.rig import AindBehaviorRigModel
 from pydantic import Field
 
-
 __version__ = "0.1.0"
 
 

src/aind_behavior_device_water_tuner/task_logic.py

@@ -25,10 +25,10 @@ class AindBehaviorWaterTunerParameters(TaskParameters):
         title="Repeat count",
     )
 
+
 class AindBehaviorWaterTunerTaskLogic(AindBehaviorTaskLogicModel):
     """WaterTuner operation control model that is used to run a calibration data acquisition workflow"""
 
     name: str = Field(default="WaterValveCalibrationLogic", title="Task name")
     version: Literal[__version__] = __version__
     task_parameters: AindBehaviorWaterTunerParameters = Field(..., title="Task parameters", validate_default=True)
-