black formatting

datajoint · Mar 25, 2024 · 83c68d5 · 83c68d5
1 parent 1f1831c
commit 83c68d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/element_moseq/moseq_train.py b/element_moseq/moseq_train.py
@@ -227,7 +227,7 @@ def make(self, key):
 
         if pose_estimation_method == "deeplabcut":
             setup_project(
-                project_dir = kpms_project_output_dir.as_posix(),
+                project_dir=kpms_project_output_dir.as_posix(),
                 deeplabcut_config=(kpset_dir / "config.yaml")
                 or (kpset_dir / "config.yml"),
             )

diff --git a/notebooks/tutorial.ipynb b/notebooks/tutorial.ipynb
@@ -3614,7 +3614,7 @@
     "full_fit_key = {\n",
     "    **pca_task_key,\n",
     "    \"full_latent_dim\": 4,\n",
-    "    \"full_kappa\": 10000.,\n",
+    "    \"full_kappa\": 10000.0,\n",
     "    \"full_num_iterations\": 5,\n",
     "    \"full_fit_desc\": \"Fitting task with kappa = 10000 ms\",\n",
     "}\n",
@@ -3638,7 +3638,7 @@
     "full_fitting_key_2 = {\n",
     "    **pca_task_key,\n",
     "    \"full_latent_dim\": 4,\n",
-    "    \"full_kappa\": 5000.,\n",
+    "    \"full_kappa\": 5000.0,\n",
     "    \"full_num_iterations\": 5,\n",
     "    \"full_fit_desc\": \"Fitting task with kappa = 5000 ms\",\n",
     "}\n",