linting

src/aind_behavior_video_transformation/etl.py

@@ -21,7 +21,7 @@
 )
 from aind_behavior_video_transformation.transform_videos import (
     CompressionRequest,
-    convert_video
+    convert_video,
 )
 
 
@@ -50,8 +50,7 @@ class BehaviorVideoJobSettings(BasicJobSettings):
         description="Run compression in parallel or sequentially.",
     )
     ffmpeg_thread_cnt: int = Field(
-        default=0,
-        description="Number of threads per ffmpeg compression job."
+        default=0, description="Number of threads per ffmpeg compression job."
     )
 
 
@@ -76,7 +75,7 @@ class BehaviorVideoJob(GenericEtl[BehaviorVideoJobSettings]):
 
     def _run_compression(
         self,
-        convert_video_args: list[tuple[Path, Path, tuple[str, str] | None]]
+        convert_video_args: list[tuple[Path, Path, tuple[str, str] | None]],
     ) -> None:
         """
         Runs CompressionRequests at the specified paths.
@@ -87,8 +86,11 @@ def _run_compression(
             num_jobs = len(convert_video_args)
             with ProcessPoolExecutor(max_workers=num_jobs) as executor:
                 jobs = [
-                    executor.submit(convert_video, *params,
-                                    self.job_settings.ffmpeg_thread_cnt)
+                    executor.submit(
+                        convert_video,
+                        *params,
+                        self.job_settings.ffmpeg_thread_cnt,
+                    )
                     for params in convert_video_args
                 ]
                 for job in as_completed(jobs):

src/aind_behavior_video_transformation/filesystem.py

@@ -4,8 +4,6 @@
 from os.path import relpath
 from pathlib import Path
 
-from aind_behavior_video_transformation.transform_videos import convert_video
-
 
 def likely_video_file(file: Path) -> bool:
     """
@@ -130,4 +128,4 @@ def transform_directory(
                 out_path = dst_dir / file_name
                 symlink(file_path, out_path)
 
-    return convert_video_args
+    return convert_video_args

src/aind_behavior_video_transformation/transform_videos.py

@@ -187,7 +187,7 @@ def convert_video(
     video_path: Path,
     output_dir: Path,
     arg_set: Optional[Tuple[str, str]],
-    ffmpeg_thread_cnt: int = 0
+    ffmpeg_thread_cnt: int = 0,
 ) -> Path:
     """
     Converts a video to a specified format using ffmpeg.

tests/test_transform_videos.py

@@ -43,9 +43,6 @@ def helper_run_compression_job(job_settings, mock_time):
 class TestBehaviorVideoJob(unittest.TestCase):
     """Test methods in BehaviorVideoJob class."""
 
-    # NOTE:
-    # Test suite does not run yet.
-    # Resolving lint errors first.
     test_data_path = Path("tests/test_video_in_dir").resolve()
     dummy_response = JobResponse(
         status_code=200,