Lint fixes

VirtualBrainLab · Jan 30, 2024 · e0b100c · e0b100c
1 parent 7951265
commit e0b100c
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/src/ephys_link/platforms/new_scale_pathfinder_handler.py b/src/ephys_link/platforms/new_scale_pathfinder_handler.py
@@ -8,6 +8,7 @@
 from __future__ import annotations
 
 import json
+from sys import exit
 from typing import TYPE_CHECKING
 from urllib import request
 from urllib.error import URLError
@@ -197,9 +198,9 @@ def _get_shank_count(self, manipulator_id: str) -> com.ShankCountOutputData:
             if probe["Id"] == manipulator_id:
                 if "ShankCount" in probe:
                     return com.ShankCountOutputData(probe["ShankCount"], "")
-                else:
-                    # Default to 1.0 if shank count is not found
-                    return com.ShankCountOutputData(1, "")
+
+                # Default to 1.0 if shank count is not found
+                return com.ShankCountOutputData(1, "")
 
         return com.ShankCountOutputData(-1, "Unable to find manipulator")
 
@@ -219,11 +220,11 @@ def _bypass_calibration(self, manipulator_id: str) -> str:
         return ""
 
     def _set_can_write(
-            self,
-            manipulator_id: str,
-            can_write: bool,
-            hours: float,
-            sio: socketio.AsyncServer,
+        self,
+        manipulator_id: str,
+        can_write: bool,
+        hours: float,
+        sio: socketio.AsyncServer,
     ) -> com.StateOutputData:
         raise NotImplementedError