fixing error in declaration

src/aind_smartspim_data_transformation/compress/png_to_zarr.py

@@ -545,7 +545,6 @@ def smartspim_channel_zarr_writer(
     logger: logging.Logger,
     stack_name: str,
     writing_options,
-    client,
 ):
     """
     Writes a fused SmartSPIM channel in OMEZarr
@@ -597,7 +596,7 @@ def smartspim_channel_zarr_writer(
     image_data = image_data.rechunk(final_chunksize)
     image_data = pad_array_n_d(arr=image_data)
 
-    print(f"About to write {image_data} in {output_path} with stack name {stack_name}")
+    print(f"Writing {image_data} from {stack_name} in {output_path}")
 
     # Creating Zarr dataset
     store = parse_url(path=output_path, mode="w").store

src/aind_smartspim_data_transformation/smartspim_job.py

@@ -5,15 +5,14 @@
 import sys
 from datetime import datetime
 from pathlib import Path
-from typing import Iterator, List, Literal, Optional
+from typing import Iterator, List, Optional
 
 from aind_data_transformation.core import (
     BasicJobSettings,
     GenericEtl,
     JobResponse,
     get_parser,
 )
-from distributed.utils import silence_logging_cmgr
 from numcodecs.blosc import Blosc
 from pydantic import Field
 
@@ -154,10 +153,7 @@ def _compress_and_write_channels(
                 )
 
         except Exception as e:
-            print(f"Error converting array: {stack_name} {e}")
-        # Closing client
-        # with silence_logging_cmgr(logging.CRITICAL):
-        #     client.shutdown()
+            print(f"Error converting array: {e}")
 
         try:
             _cleanup(deployment)

tests/test_smartspim_job.py

@@ -81,5 +81,6 @@ def test_compress_and_write_channels(self):
         """Tests SmartSPIM compression and zarr writing"""
         pass
 
+
 if __name__ == "__main__":
     unittest.main()