Merge pull request #172 from sb-ai-lab/dev/matching-n-neighbors

Dev/matching n neighbors

Author: TonyKatkov89

hypex/analyzers/ab.py

@@ -4,13 +4,7 @@
 from typing import Any
 
 from ..comparators import TTest, UTest
-from ..dataset import (
-    Dataset,
-    ExperimentData,
-    StatisticRole,
-    TargetRole,
-    TreatmentRole,
-)
+from ..dataset import Dataset, ExperimentData, StatisticRole, TargetRole, TreatmentRole
 from ..experiments.base import Executor
 from ..extensions.statsmodels import MultiTest, MultitestQuantile
 from ..utils import (

hypex/comparators/abstract.py

@@ -288,7 +288,7 @@ def _split_for_cross_mode(
         compared_data = cls._split_ds_into_columns(data=compared_data)
 
         return baseline_data, compared_data
-    
+
     @classmethod
     def _split_for_matched_pairs_mode(
         cls,
@@ -408,7 +408,7 @@ def execute(self, data: ExperimentData) -> ExperimentData:
         group_field_data = fields["group_field"]
         target_fields_data = fields["target_fields"]
         baseline_field_data = fields["baseline_field"]
-        
+
         self.key = str(
             target_fields_data.columns[0]
             if len(target_fields_data.columns) == 1

hypex/dataset/__init__.py

@@ -5,28 +5,28 @@
 from .abstract import DatasetBase
 from .dataset import Dataset, DatasetAdapter, ExperimentData
 from .roles import (
-    ABCRole,
-    AdditionalGroupingRole,
-    AdditionalMatchingRole,
-    AdditionalPreTargetRole,
-    AdditionalTargetRole,
-    AdditionalTreatmentRole,
-    ConstGroupRole,
-    DefaultRole,
-    FeatureRole,
-    FilterRole,
-    GroupingRole,
-    InfoRole,
-    PreTargetRole,
-    StatisticRole,
-    StratificationRole,
-    TargetRole,
-    TempGroupingRole,
-    TempRole,
-    TempTargetRole,
-    TempTreatmentRole,
-    TreatmentRole,
-    default_roles,
+                    ABCRole,
+                    AdditionalGroupingRole,
+                    AdditionalMatchingRole,
+                    AdditionalPreTargetRole,
+                    AdditionalTargetRole,
+                    AdditionalTreatmentRole,
+                    ConstGroupRole,
+                    DefaultRole,
+                    FeatureRole,
+                    FilterRole,
+                    GroupingRole,
+                    InfoRole,
+                    PreTargetRole,
+                    StatisticRole,
+                    StratificationRole,
+                    TargetRole,
+                    TempGroupingRole,
+                    TempRole,
+                    TempTargetRole,
+                    TempTreatmentRole,
+                    TreatmentRole,
+                    default_roles,
 )
 
 __all__ = [

hypex/dataset/abstract.py

@@ -9,11 +9,7 @@
 
 from ..utils import BackendsEnum, RoleColumnError
 from .backends import PandasDataset
-from .roles import (
-    ABCRole,
-    DefaultRole,
-    default_roles,
-)
+from .roles import ABCRole, DefaultRole, default_roles
 
 
 def parse_roles(roles: dict) -> dict[str | int] | ABCRole:

hypex/dataset/backends/abstract.py

@@ -1,14 +1,7 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import (
-    Any,
-    Callable,
-    Iterable,
-    Literal,
-    Sequence,
-    Sized,
-)
+from typing import Any, Callable, Iterable, Literal, Sequence, Sized
 
 from ...utils import AbstractMethodError, FromDictTypes
 

hypex/dataset/backends/pandas_backend.py

@@ -1,14 +1,7 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import (
-    Any,
-    Callable,
-    Iterable,
-    Literal,
-    Sequence,
-    Sized,
-)
+from typing import Any, Callable, Iterable, Literal, Sequence, Sized
 
 import numpy as np
 import pandas as pd  # type: ignore

hypex/dataset/dataset.py

@@ -3,13 +3,7 @@
 import warnings
 from collections.abc import Iterable
 from copy import deepcopy
-from typing import (
-    Any,
-    Callable,
-    Hashable,
-    Literal,
-    Sequence,
-)
+from typing import Any, Callable, Hashable, Literal, Sequence
 
 import pandas as pd  # type: ignore
 

hypex/dataset/roles.py

@@ -2,12 +2,7 @@
 
 from abc import ABC
 
-from ..utils import (
-    CategoricalTypes,
-    DefaultRoleTypes,
-    RoleNameType,
-    TargetRoleTypes,
-)
+from ..utils import CategoricalTypes, DefaultRoleTypes, RoleNameType, TargetRoleTypes
 
 
 class ABCRole(ABC):

hypex/experiments/__init__.py

@@ -1,7 +1,4 @@
-from .base import (
-    Experiment,
-    OnRoleExperiment,
-)
+from .base import Experiment, OnRoleExperiment
 from .base_complex import CycledExperiment, GroupExperiment
 
 __all__ = ["CycledExperiment", "Experiment", "GroupExperiment", "OnRoleExperiment"]

hypex/experiments/base.py

@@ -3,11 +3,7 @@
 from copy import deepcopy
 from typing import Any, Iterable, Sequence
 
-from ..dataset import (
-    ABCRole,
-    ExperimentData,
-    TempTargetRole,
-)
+from ..dataset import ABCRole, ExperimentData, TempTargetRole
 from ..executor import Executor
 from ..utils import ExperimentDataEnum