refactor into causalpy class structure (not updated notebook yet)

Author: drbenvincent

causalpy/__init__.py

@@ -19,14 +19,14 @@
 
 from .data import load_data
 from .experiments.diff_in_diff import DifferenceInDifferences
+from .experiments.graded_intervention_its import GradedInterventionTimeSeries
 from .experiments.instrumental_variable import InstrumentalVariable
 from .experiments.interrupted_time_series import InterruptedTimeSeries
 from .experiments.inverse_propensity_weighting import InversePropensityWeighting
 from .experiments.prepostnegd import PrePostNEGD
 from .experiments.regression_discontinuity import RegressionDiscontinuity
 from .experiments.regression_kink import RegressionKink
 from .experiments.synthetic_control import SyntheticControl
-from .experiments.transfer_function_its import TransferFunctionITS
 from .transforms import (
     AdstockTransform,
     DiscreteLag,
@@ -42,10 +42,11 @@
 __all__ = [
     "__version__",
     "AdstockTransform",
-    "DifferenceInDifferences",
     "create_causalpy_compatible_class",
+    "DifferenceInDifferences",
     "DiscreteLag",
     "GeometricAdstock",
+    "GradedInterventionTimeSeries",
     "HillSaturation",
     "InstrumentalVariable",
     "InterruptedTimeSeries",
@@ -61,6 +62,5 @@
     "SaturationTransform",
     "skl_models",
     "SyntheticControl",
-    "TransferFunctionITS",
     "Treatment",
 ]