added retrain code

epdb/views.py

@@ -955,6 +955,12 @@ def package_model(request, package_uuid, model_uuid):
                 ]
                 dispatch(current_user, evaluate_model, current_model.pk, multigen, eval_package_ids)
 
+                return redirect(current_model.url)
+            elif hidden == "retrain":
+                from .tasks import dispatch, retrain
+
+                dispatch(current_user, retrain, current_model.pk)
+
                 return redirect(current_model.url)
             else:
                 return HttpResponseBadRequest()

templates/modals/objects/retrain_model_modal.html

@@ -1,5 +1,11 @@
-<div class="modal fade" tabindex="-1" id="retrain_model_modal" role="dialog" aria-labelledby="retrain_model_modal"
+<div
-     aria-hidden="true">
+  class="modal fade"
+  tabindex="-1"
+  id="retrain_model_modal"
+  role="dialog"
+  aria-labelledby="retrain_model_modal"
+  aria-hidden="true"
+>
   <div class="modal-dialog modal-lg">
     <div class="modal-content">
       <div class="modal-header">
@@ -10,34 +16,40 @@
         <h4 class="modal-title">Retrain Model</h4>
       </div>
       <div class="modal-body">
-                <form id="retrain_model_form" accept-charset="UTF-8" action="{{ meta.current_package.url }}/model"
+        <form
-                      data-remote="true" method="post">
+          id="retrain_model_form"
+          accept-charset="UTF-8"
+          action="{{ meta.current_object.url }}"
+          data-remote="true"
+          method="post"
+        >
           <div class="jumbotron">
-                        To reflect changes in the rule or data packages, you can use the "Retrain" button,
+            To reflect changes in the rule or data packages, you can use the
-                        to let the model reflect the changes without creating a new model.
+            "Retrain" button, to let the model reflect the changes without
-                        While the model is retraining, it will be unavailable for prediction.
+            creating a new model. While the model is retraining, it will be
+            unavailable for prediction.
           </div>
           {% csrf_token %}
-                    <input type="hidden" name="action" value="retrain">
+          <input type="hidden" name="hidden" value="retrain" />
         </form>
       </div>
       <div class="modal-footer">
-                <a id="retrain_model_form_submit" class="btn btn-primary" href="#">Retrain</a>
+        <a id="retrain_model_form_submit" class="btn btn-primary" href="#"
-                <button type="button" class="btn btn-default" data-dismiss="modal">Cancel</button>
+          >Retrain</a
+        >
+        <button type="button" class="btn btn-default" data-dismiss="modal">
+          Cancel
+        </button>
       </div>
     </div>
   </div>
 </div>
 
 <script>
-
   $(function () {
-
+    $("#retrain_model_form_submit").on("click", function (e) {
-        $('#retrain_model_form_submit').on('click', function (e) {
       e.preventDefault();
-            $('#retrain_model_form').submit();
+      $("#retrain_model_form").submit();
     });
-
   });
-
 </script>