SIENTIAPDE-1182
Update tests.ipynb and values.yaml for improved functionality and versioning - Updated execution count in tests.ipynb for reproducibility. - Modified DataFrame creation in tests.ipynb to include a timestamp column and save to CSV. - Changed image tag in values.yaml from "0.4.4" to "0.4.5" for versioning. - Updated GITHUB_BRANCH in values.yaml to reflect the latest branch adjustments.
This commit is contained in:
@@ -39,6 +39,7 @@ class MLFlowRepository():
|
||||
"""
|
||||
|
||||
try:
|
||||
|
||||
return {
|
||||
'success': True,
|
||||
'content': self.model_serving.get_cached_transform(
|
||||
@@ -67,12 +68,15 @@ class MLFlowRepository():
|
||||
- dict: A dictionary containing the predicted data.
|
||||
"""
|
||||
try:
|
||||
|
||||
input_index = data.index
|
||||
start_time = datetime.now()
|
||||
data = self.model_serving.get_cached_predict(
|
||||
model_name, data, model_retention)
|
||||
|
||||
end_time = datetime.now()
|
||||
data = pd.DataFrame(data, columns=['prediction'])
|
||||
data.index = input_index
|
||||
data['response_time'] = (end_time - start_time).total_seconds()
|
||||
|
||||
return {
|
||||
|
||||
Reference in New Issue
Block a user