Generic input interface for Algo base class (#5602)

Making Algo base to have more generic inputs.

Signed-off-by: myron <amyronenko@nvidia.com>

Author: myron

monai/auto3dseg/algo_gen.py

@@ -9,7 +9,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Optional
 
 from monai.transforms import Randomizable
 
@@ -24,23 +23,12 @@ def set_data_stats(self, *args, **kwargs):
         """Provide dataset (and summaries) so that the model creation can depend on the input datasets."""
         pass
 
-    def train(self, params: dict):
-        """
-        Read training/validation data and output a model.
-
-        Args:
-            params: key-value pairs of input parameters for the training pipeline.
-        """
+    def train(self, *args, **kwargs):
+        """Read training/validation data and output a model."""
         pass
 
-    def predict(self, predict_files: list, predict_params: Optional[dict] = None):
-        """
-        Read test data and output model predictions.
-
-        Args:
-            predict_files: list of files for the predicting pipeline.
-            predict_params: key-value pairs of input parameters for the predicting pipeline.
-        """
+    def predict(self, *args, **kwargs):
+        """Read test data and output model predictions."""
         pass
 
     def get_score(self, *args, **kwargs):