templates.general-mlflow.training.config.py.vm Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of foundation-mda Show documentation

Model driven architecture artifacts for aiSSEMBLE

The newest version!

"""
Configurations for this pipeline, read from the pipeline properties file.

GENERATED STUB CODE - PLEASE ***DO*** MODIFY

Originally generated from: ${templateName} 
"""

from krausening.properties import PropertyManager


class PipelineConfig():
    """
    Configurations for this pipeline, read from the pipeline properties file.
    """

    def __init__(self):
        self.properties = PropertyManager.get_instance().get_properties('pipeline.properties')


    def mlflow_tracking_uri(self):
        """
        Returns the directory for tracking MLflow training runs.
        """
        return self.properties['mlflow_tracking_uri']


    def kafka_server(self):
        """
        Returns the Kafka server host:port value.
        """
        return self.properties['kafka_server']


    #if (${containsOnnx})
    def onnx_model_directory(self):
        """
        Returns the directory's name to save onnx model on S3.
        """
        return self.properties['onnx_model_directory']
    #end