Merge pull request #120 from AllenNeuralDynamics:hotfix-distributions…

…-annotated Fix distributions class discriminator type signature
AllenNeuralDynamics · Nov 9, 2024 · 5acdbfe · 5acdbfe
2 parents 3ec4c26 + 489df1c
commit 5acdbfe
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/aind_behavior_services/__init__.py b/src/aind_behavior_services/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.8.4"
+__version__ = "0.8.5"
 
 from .rig import AindBehaviorRigModel  # noqa: F401
 from .session import AindBehaviorSessionModel  # noqa: F401

diff --git a/src/aind_behavior_services/rig/__init__.py b/src/aind_behavior_services/rig/__init__.py
@@ -25,7 +25,8 @@ class VideoWriterFfmpeg(BaseModel):
         description="Output arguments",
     )
     input_arguments: str = Field(
-        default="-v verbose -colorspace bt709 -color_primaries bt709 -color_range full -color_trc linear", description="Input arguments"
+        default="-v verbose -colorspace bt709 -color_primaries bt709 -color_range full -color_trc linear",
+        description="Input arguments",
     )
 
 

diff --git a/src/aind_behavior_services/task_logic/distributions.py b/src/aind_behavior_services/task_logic/distributions.py
@@ -206,9 +206,9 @@ class PdfDistribution(DistributionBase):
             BetaDistribution,
             GammaDistribution,
             PdfDistribution,
-        ]
+        ],
+        Field(discriminator="family", title="Distribution", description="Available distributions"),
     ],
-    Field(discriminator="family", title="Distribution", description="Available distributions"),
 )
 
 DistributionParameters = TypeAliasType(