diff --git a/pyproject.toml b/pyproject.toml index 56dc8f1..569643d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -16,7 +16,9 @@ classifiers = [ ] dynamic = ["version"] -dependencies = [] +dependencies = [ + "deepdisc@git+https://github.com/lincc-frameworks/deepdisc.git", +] # On a mac, install optional dependencies with `pip install '.[dev]'` (include the single quotes) [project.optional-dependencies] diff --git a/src/rail/deepdisc/__init__.py b/src/rail/deepdisc/__init__.py index 0588fc3..f9a0d32 100644 --- a/src/rail/deepdisc/__init__.py +++ b/src/rail/deepdisc/__init__.py @@ -1,3 +1 @@ -from ._version import __version__ - from rail.estimation.algos.deepdisc import * diff --git a/src/rail/estimation/algos/deepdisc.py b/src/rail/estimation/algos/deepdisc.py index e69de29..9602309 100644 --- a/src/rail/estimation/algos/deepdisc.py +++ b/src/rail/estimation/algos/deepdisc.py @@ -0,0 +1,24 @@ +from rail.estimation.estimator import CatEstimator, CatInformer + +class DeepDiscInformer(CatInformer): + """Placeholder for informer stage class""" + + name = "DeepDiscInformer" + config_options = CatInformer.config_options.copy() + + def __init__(self, args, comm=None): + CatInformer.__init__(self, args, comm=comm) + + def run(self): + pass + + +class DeepDiscEstimator(CatEstimator): + name = "DeepDiscEstimator" + config_options = CatEstimator.config_options.copy() + + def __init__(self, args, comm=None): + CatEstimator.__init__(self, args, comm=comm) + + def _process_chunk(self, start, end, data, first): + pass diff --git a/tests/deepdisc/test_placeholder.py b/tests/deepdisc/test_placeholder.py new file mode 100644 index 0000000..3ada1ee --- /dev/null +++ b/tests/deepdisc/test_placeholder.py @@ -0,0 +1,2 @@ +def test_placeholder(): + assert True