SIENTIAPDE-1222
SIENTIAPDE-1222 Update MLFlow methods to include metadata parameter - Modified the transform and predict method calls in the MLFlow class to include a new 'metadata' parameter, enhancing the functionality and data handling capabilities of the model monitoring repository.
This commit is contained in:
@@ -120,7 +120,7 @@ class MLFlow(BaseActivity):
|
|||||||
|
|
||||||
# Request transformation from MLFlow model
|
# Request transformation from MLFlow model
|
||||||
response_data = self.model_monitoring_repository.transform(
|
response_data = self.model_monitoring_repository.transform(
|
||||||
model_name, data, model_config
|
model_name, data, model_config, metadata
|
||||||
)
|
)
|
||||||
|
|
||||||
self.debug(
|
self.debug(
|
||||||
@@ -176,7 +176,7 @@ class MLFlow(BaseActivity):
|
|||||||
|
|
||||||
# Request prediction from MLFlow model
|
# Request prediction from MLFlow model
|
||||||
response_data = self.model_monitoring_repository.predict(
|
response_data = self.model_monitoring_repository.predict(
|
||||||
model_name, data, model_config
|
model_name, data, model_config, metadata
|
||||||
)
|
)
|
||||||
|
|
||||||
self.debug(
|
self.debug(
|
||||||
|
|||||||
Reference in New Issue
Block a user