chore(brick): fix generator annotation

brick/api/client.py

@@ -59,7 +59,7 @@ async def close(self):
 
 async def fetch_sessions_generator(
     api_client: ApiClient, logger: logging.Logger | None = None
-) -> AsyncGenerator[Session]:
+) -> AsyncGenerator[Session, None, None]:
     if logger:
         logger.info(f"Requesting session identifiers from API")
 

brick/api/core/config.py

@@ -82,8 +82,10 @@ def get_mongodb_secret_pwd(cls, v: str):
     @field_validator("GENOMAD_DATABASE", mode="after")
     def check_genomad_database(cls, v: Path):
         if v and not v.exists():
-            logging.warn(f"geNomad database directory not found! ({v})")
-            logging.warn(
+            logging.warning(
+                f"geNomad database directory not found! ({v})"
+            )  # tests run on action runner must avoid pulling database
+            logging.warning(
                 f"Attempts to execute `genomad` will fail in the `process_genomad_ring` worker!"
             )
         return v

brick/api/main.py

@@ -29,8 +29,8 @@ def init_working_directory(settings: Settings):
 
     # Create working directory
     if not settings.WORK_DIRECTORY.exists():
-        logging.warn(f"Working directory does not exist: {settings.WORK_DIRECTORY}")
-        logging.warn(
+        logging.warning(f"Working directory does not exist: {settings.WORK_DIRECTORY}")
+        logging.warning(
             f"Attempting to create working directory path for server operations..."
         )
 

brick/api/tasks.py

@@ -843,6 +843,6 @@ def check_or_update_label_ring(
 
 
 @contextlib.contextmanager
-def create_tmp_directory(root_dir: str = None) -> Generator[Path]:
+def create_tmp_directory(root_dir: str = None) -> Generator[Path, None, None]:
     with tempfile.TemporaryDirectory(dir=root_dir) as temp_dir:
         yield Path(temp_dir)

brick/rings.py

@@ -279,7 +279,7 @@ def parse_blastn_output(
 
 
 # Generator function
-def parse_aggregated_genomad_output(file: Path) -> Generator[GenomadEntry]:
+def parse_aggregated_genomad_output(file: Path) -> Generator[GenomadEntry, None, None]:
     """
     Parses the aggregated_classification output from a sliced genome file
     """
@@ -306,7 +306,7 @@ def extract_genomad_contiguous_segments(
     min_segment_length: int,  # should be a multiple of slice length
     prediction_classes: List[GenomadPredictionClass],
     segment_type: RingSegmentType,
-) -> Generator[RingSegment] | Generator[LabelSegment]:
+) -> Generator[RingSegment, None, None] | Generator[LabelSegment, None, None]:
     """Extracts segments of high probabilty for each prediction class with a minimum total length for label or annotation rings"""
 
     segments = []