update format

Aharoni-Lab · Nov 1, 2024 · 97a57a2 · 97a57a2
1 parent 6dcd57a
commit 97a57a2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/miniscope_io/cli/update.py b/miniscope_io/cli/update.py
@@ -58,9 +58,8 @@ def update(port: str, target: str, value: int, device_id: int, restart: bool) ->
     if target and value:
         DevUpdate(port=port, target=target, value=value, device_id=device_id)
     elif restart:
-        DevUpdate(port=port,
-                  target="DEVICE",
-                  value=DeviceCommand.RESTART.value,
-                  device_id=device_id)
+        DevUpdate(
+            port=port, target="DEVICE", value=DeviceCommand.RESTART.value, device_id=device_id
+        )
     else:
         raise click.UsageError("Either --target with --value or --restart must be specified.")
diff --git a/tests/test_models/test_model_devupdate.py b/tests/test_models/test_model_devupdate.py
@@ -45,4 +45,4 @@ def test_invalid_port():
 
 def test_device_command(mock_serial_ports):
     cmd = DevUpdateCommand(device_id=1, port="COM2", target="DEVICE", value=DeviceCommand.RESTART.value)
-    assert cmd.value == int(DeviceCommand.RESTART.value)
+    assert cmd.value == DeviceCommand.RESTART.value