ruff

Author: bpblanken

v03_pipeline/lib/misc/clickhouse.py

@@ -55,7 +55,7 @@ def src_path_fn(self) -> Callable:
             ClickHouseTable.ENTRIES: new_entries_parquet_path,
         }[self]
 
-    def should_load(
+    def should_load(  # noqa: PLR0911
         self,
         reference_genome: ReferenceGenome,
         dataset_type: DatasetType,

v03_pipeline/lib/tasks/exports/write_new_variants_parquet.py

@@ -15,6 +15,7 @@
 from v03_pipeline.lib.tasks.exports.fields import get_variants_export_fields
 from v03_pipeline.lib.tasks.exports.misc import (
     camelcase_array_structexpression_fields,
+    subset_sorted_transcript_consequences_fields,
     unmap_formatting_annotation_enums,
     unmap_reference_dataset_annotation_enums,
 )
@@ -26,9 +27,6 @@
     UpdateVariantAnnotationsTableWithNewSamplesTask,
 )
 from v03_pipeline.lib.tasks.write_new_variants_table import WriteNewVariantsTableTask
-from v03_pipeline.lib.tasks.exports.misc import (
-    subset_sorted_transcript_consequences_fields,
-)
 
 
 @luigi.util.inherits(BaseLoadingRunParams)

v03_pipeline/lib/tasks/exports/write_new_variants_parquet_test.py

@@ -363,7 +363,6 @@ def test_mito_write_new_variants_parquet_test(
         export_json[0]['sortedTranscriptConsequences'] = [
             export_json[0]['sortedTranscriptConsequences'][0],
         ]
-        print(export_json[0]['sortedTranscriptConsequences'])
         self.assertEqual(
             export_json,
             [
@@ -418,7 +417,7 @@ def test_mito_write_new_variants_parquet_test(
                             'biotype': 'Mt_tRNA',
                             'consequenceTerms': ['non_coding_transcript_exon_variant'],
                             'lofFilters': None,
-                        }
+                        },
                     ],
                 },
             ],