add json.dumps in requests nlp train

weni-ai · Apr 15, 2021 · fe52bdc · fe52bdc
1 parent dee3d7c
commit fe52bdc
Showing 1 changed file with 9 additions and 24 deletions.
diff --git a/bothub/common/models.py b/bothub/common/models.py
@@ -431,31 +431,16 @@ def validate_if_can_run_automatic_evaluate(self, language: str) -> None:
 
     def request_nlp_train(self, user_authorization, data):
         try:  # pragma: no cover
-            if data.get("repository_version"):
-                r = requests.post(  # pragma: no cover
-                    "{}train/".format(
-                        self.nlp_server
-                        if self.nlp_server
-                        else settings.BOTHUB_NLP_BASE_URL
-                    ),
-                    data={"repository_version": data.get("repository_version")},
-                    headers={
-                        "Authorization": "Bearer {}".format(user_authorization.uuid)
-                    },
-                )
-            else:
-                r = requests.post(  # pragma: no cover
-                    "{}train/".format(
-                        self.nlp_server
-                        if self.nlp_server
-                        else settings.BOTHUB_NLP_BASE_URL
-                    ),
-                    data={},
-                    headers={
-                        "Authorization": "Bearer {}".format(user_authorization.uuid)
-                    },
-                )
+            url = f"{self.nlp_server if self.nlp_server else settings.BOTHUB_NLP_BASE_URL}train/"
+            data = {
+                "repository_version": data.get("repository_version")
+            }
+            headers = {"Authorization": f"Bearer {user_authorization.uuid}"}
+
+            r = requests.post(url, data=json.dumps(data), headers=headers)
+
             return r  # pragma: no cover
+
         except requests.exceptions.ConnectionError:  # pragma: no cover
             raise APIException(  # pragma: no cover
                 {"status_code": status.HTTP_503_SERVICE_UNAVAILABLE},