From d14b7b021121910a86810caa50c4fb1d1228f257 Mon Sep 17 00:00:00 2001 From: Declan McNamara Date: Mon, 3 Feb 2025 13:48:41 -0500 Subject: [PATCH] linting --- case_studies/redshift/artifacts/redshift_dc2.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/case_studies/redshift/artifacts/redshift_dc2.py b/case_studies/redshift/artifacts/redshift_dc2.py index 785df26da..5e621dfb6 100644 --- a/case_studies/redshift/artifacts/redshift_dc2.py +++ b/case_studies/redshift/artifacts/redshift_dc2.py @@ -1,14 +1,11 @@ - import logging import multiprocessing import pathlib from pathlib import Path - - import torch -from bliss.surveys.dc2 import DC2DataModule, map_nested_dicts, split_list, unpack_dict, split_tensor +from bliss.surveys.dc2 import DC2DataModule, map_nested_dicts, split_list, split_tensor, unpack_dict class RedshiftDC2DataModule(DC2DataModule): @@ -19,10 +16,7 @@ def __init__( *args, **kwargs, ): - super().__init__( - *args, - **kwargs - ) + super().__init__(*args, **kwargs) self.dc2_image_dir = Path(self.dc2_image_dir) self.dc2_cat_path = Path(self.dc2_cat_path) @@ -159,5 +153,3 @@ def generate_cached_data(self, naming_info: tuple): with open(cached_data_file_path, "wb") as cached_data_file: torch.save(tmp_data_cached, cached_data_file) data_count += 1 - - \ No newline at end of file