From 1c85f1a0bcf15181d72c477d95f6a9fac67bda99 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Wed, 15 Jul 2020 16:39:48 +0200
Subject: [PATCH] abstract model class is available now for mlair
---
mlair/__init__.py | 3 ++-
mlair/model_modules/__init__.py | 2 ++
2 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/mlair/__init__.py b/mlair/__init__.py
index 8ff82e6c..bf1a5fdd 100644
--- a/mlair/__init__.py
+++ b/mlair/__init__.py
@@ -4,9 +4,10 @@ __version_info__ = {
'micro': 0,
}
-from mlair.run_modules import *
+from mlair.run_modules import RunEnvironment, ExperimentSetup, PreProcessing, ModelSetup, Training, PostProcessing
from mlair.workflows import DefaultWorkflow, Workflow
from mlair.run_script import run
+from mlair.model_modules import AbstractModelClass
def get_version():
diff --git a/mlair/model_modules/__init__.py b/mlair/model_modules/__init__.py
index 35f40608..ea2067bd 100644
--- a/mlair/model_modules/__init__.py
+++ b/mlair/model_modules/__init__.py
@@ -1 +1,3 @@
"""Collection of all modules that are related to a model."""
+
+from .model_class import AbstractModelClass
--
GitLab