From 792f658cc87733983b351f66d9bc71765e7fe2ba Mon Sep 17 00:00:00 2001
From: Daniel Engbert <78389608+dan-scale@users.noreply.github.com>
Date: Mon, 3 May 2021 22:07:59 -0400
Subject: [PATCH] Add self_serve arg to project creation. (#35)

* Add self_serve arg to project creation.

* Formatting fixes
---
 scaleapi/__init__.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/scaleapi/__init__.py b/scaleapi/__init__.py
index 7d80431..79ff89f 100644
--- a/scaleapi/__init__.py
+++ b/scaleapi/__init__.py
@@ -504,7 +504,11 @@ def get_batches(
             has_more = batches.has_more
 
     def create_project(
-        self, project_name: str, task_type: TaskType, params: Dict = None
+        self,
+        project_name: str,
+        task_type: TaskType,
+        params: Dict = None,
+        self_serve: bool = False,
     ) -> Project:
         """Creates a new project.
         https://docs.scale.com/reference#project-creation
@@ -524,7 +528,12 @@ def create_project(
             Project: [description]
         """
         endpoint = "projects"
-        payload = dict(type=task_type.value, name=project_name, params=params)
+        payload = dict(
+            type=task_type.value,
+            name=project_name,
+            params=params,
+            self_serve=self_serve,
+        )
         projectdata = self.api.post_request(endpoint, body=payload)
         return Project(projectdata, self)