diff --git a/src/spyglass/behavior/moseq.py b/src/spyglass/behavior/moseq.py index 6c0d9e38a..26d6e5805 100644 --- a/src/spyglass/behavior/moseq.py +++ b/src/spyglass/behavior/moseq.py @@ -23,7 +23,11 @@ class MoseqModelParams(SpyglassMixin, dj.Manual): """ def make_training_extension_params( - self, model_key:dict, num_epochs:int, new_name:str=None, skip_duplicates:bool=False + self, + model_key: dict, + num_epochs: int, + new_name: str = None, + skip_duplicates: bool = False, ): """Method to create a new set of model parameters for extending training @@ -37,7 +41,7 @@ def make_training_extension_params( name for the new model parameters, by default None skip_duplicates : bool, optional whether to skip duplicates, by default False - + Returns ------- dict