Azure predict proba issue
Run project with use_a2ml_hub: true in config.yml
train and deploy model
then run predict with threshold
target will contain class numbers, not actual classes. For iris:
prediction_id,sepal_length,sepal_width,petal_length,petal_width,class,proba_0,proba_1,proba_2
e37f8e1b-c70e-4ef3-9cec-fe914d963720,6.0,2.2,5.0,1.5,2,0.0006133182633167796,0.02319485741752154,0.9761918243191617

diff --git a/a2ml/api/auger/impl/cloud/pipeline.py b/a2ml/api/auger/impl/cloud/pipeline.py
index 18b3513..9f85d52 100644
--- a/a2ml/api/auger/impl/cloud/pipeline.py
+++ b/a2ml/api/auger/impl/cloud/pipeline.py
@@ -16,7 +16,7 @@ class AugerPipelineApi(AugerBaseApi):
             ['creating_files', 'packaging', 'deploying'])
 
     def remove(self, trial_id):
-        return self._call_update({'id': trial_id, 'status': 'undeploying'})
+        return self._call_update({'id': trial_id, 'status': 'undeploying'}, progress=['ready','packaged_with_error'])
 
     def predict(self, records, features, threshold=None, file_url=None, predicted_at=None):
         if self.object_id is None:
diff --git a/a2ml/tasks_queue/tasks_hub_api.py b/a2ml/tasks_queue/tasks_hub_api.py
index 9225952..812b161 100644
--- a/a2ml/tasks_queue/tasks_hub_api.py
+++ b/a2ml/tasks_queue/tasks_hub_api.py
@@ -397,11 +397,14 @@ def undeploy_model_task(params):
     res = A2MLModel(ctx).undeploy(model_id = model_id)
     _update_hub_objects(ctx, params.get('provider'), params)
 
+    return res
+
 @celeryApp.task(ignore_result=True, after_return=process_task_result)
 def predict_by_model_task(params):
     from a2ml.api.utils.crud_runner import CRUDRunner
 
     ctx = _create_provider_context(params)
+    ctx = _read_hub_experiment_session(ctx, params)
 
     ctx.config.clean_changes()
     runner = CRUDRunner(ctx, "%s"%params.get('provider'), 'model')
['a2ml/tasks_queue/tasks_hub_api.py', 'a2ml/api/auger/impl/cloud/pipeline.py']