From c5002cd1fc9ad9decfcb80570b665d4ce72b29f6 Mon Sep 17 00:00:00 2001 From: Henry Date: Fri, 12 Jul 2024 14:14:41 +0200 Subject: [PATCH] :art: update type hints, remove unused import --- src/move/tasks/identify_associations.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/src/move/tasks/identify_associations.py b/src/move/tasks/identify_associations.py index b979cb5..7f1b853 100644 --- a/src/move/tasks/identify_associations.py +++ b/src/move/tasks/identify_associations.py @@ -9,7 +9,6 @@ import numpy as np import pandas as pd import torch -import torch.multiprocessing from omegaconf import OmegaConf from scipy.stats import ks_2samp, pearsonr # type: ignore from torch.utils.data import DataLoader @@ -79,11 +78,7 @@ def prepare_for_categorical_perturbation( config: MOVEConfig, interim_path: Path, baseline_dataloader: DataLoader, -) -> tuple[ - list[DataLoader], - BoolArray, - BoolArray, -]: +) -> list[DataLoader]: """ This function creates the required dataloaders and masks for further categorical association analysis. @@ -125,11 +120,7 @@ def prepare_for_continuous_perturbation( config: MOVEConfig, output_subpath: Path, baseline_dataloader: DataLoader, -) -> tuple[ - list[DataLoader], - BoolArray, - BoolArray, -]: +) -> list[DataLoader]: """ This function creates the required dataloaders and masks for further continuous association analysis.