From 39b2a23c8035fb205d232fefc19c7dba7ca549a8 Mon Sep 17 00:00:00 2001
From: Jiei-S <jiei7414@gmail.com>
Date: Sun, 11 Jun 2023 16:33:59 +0900
Subject: [PATCH 1/3] feat: dataset create

---
 README.md                  |  8 +++++---
 examples/create_dataset.py | 13 +++++++++++++
 fastlabel/__init__.py      |  4 +++-
 3 files changed, 21 insertions(+), 4 deletions(-)
 create mode 100644 examples/create_dataset.py

diff --git a/README.md b/README.md
index a36d863..c5d5c3b 100644
--- a/README.md
+++ b/README.md
@@ -2125,7 +2125,8 @@ Create a new dataset.
 dataset = client.create_dataset(
     name="Japanese Dogs",
     slug="japanese-dogs",
-    type="image"
+    type="image",
+    annotation_type="image_bbox"
 )
 ```
 
@@ -2681,10 +2682,11 @@ if __name__ == '__main__':
 ```
 
 ## Model Monitoring
+
 ### Create Request Results
-You can integrate the results of model endpoint calls, 
-which are targeted for aggregation in model monitoring, from an external source.
 
+You can integrate the results of model endpoint calls,
+which are targeted for aggregation in model monitoring, from an external source.
 
 ```python
 from datetime import datetime
diff --git a/examples/create_dataset.py b/examples/create_dataset.py
new file mode 100644
index 0000000..5c3391a
--- /dev/null
+++ b/examples/create_dataset.py
@@ -0,0 +1,13 @@
+from pprint import pprint
+
+import fastlabel
+
+client = fastlabel.Client()
+
+dataset = client.create_dataset(
+    name="Japanese Dogs",
+    slug="japanese-dogs2",
+    type="video",
+    annotation_type="image_bbox",
+)
+pprint(dataset)
diff --git a/fastlabel/__init__.py b/fastlabel/__init__.py
index 58b8edd..68760ed 100644
--- a/fastlabel/__init__.py
+++ b/fastlabel/__init__.py
@@ -33,7 +33,6 @@
 
 
 class Client:
-
     api = None
 
     def __init__(self):
@@ -3315,6 +3314,7 @@ def create_dataset(
         type: str,
         name: str,
         slug: str,
+        annotation_type: str,
     ) -> dict:
         """
         Create a dataset.
@@ -3322,12 +3322,14 @@ def create_dataset(
         type can be 'image', 'video', 'audio' (Required).
         name is name of your dataset (Required).
         slug is slug of your dataset (Required).
+        annotation_type can be 'none', 'image_bbox', 'image_segmentation' (Required).
         """
         endpoint = "datasets"
         payload = {
             "type": type,
             "name": name,
             "slug": slug,
+            "annotationType": annotation_type,
         }
         return self.api.post_request(endpoint, payload=payload)
 

From 528b0dc97003e7555f0469e8b5a715994d19c8a5 Mon Sep 17 00:00:00 2001
From: Jiei-S <jiei7414@gmail.com>
Date: Sun, 11 Jun 2023 16:35:19 +0900
Subject: [PATCH 2/3] feat: dataset create

---
 examples/create_dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/create_dataset.py b/examples/create_dataset.py
index 5c3391a..179c061 100644
--- a/examples/create_dataset.py
+++ b/examples/create_dataset.py
@@ -6,7 +6,7 @@
 
 dataset = client.create_dataset(
     name="Japanese Dogs",
-    slug="japanese-dogs2",
+    slug="japanese-dogs",
     type="video",
     annotation_type="image_bbox",
 )

From 7f5817024ecf1fdc0838783c0d430292060bf08d Mon Sep 17 00:00:00 2001
From: Jiei-S <jiei7414@gmail.com>
Date: Sun, 11 Jun 2023 16:36:18 +0900
Subject: [PATCH 3/3] feat: dataset create

---
 fastlabel/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fastlabel/__init__.py b/fastlabel/__init__.py
index 68760ed..1181145 100644
--- a/fastlabel/__init__.py
+++ b/fastlabel/__init__.py
@@ -3322,7 +3322,7 @@ def create_dataset(
         type can be 'image', 'video', 'audio' (Required).
         name is name of your dataset (Required).
         slug is slug of your dataset (Required).
-        annotation_type can be 'none', 'image_bbox', 'image_segmentation' (Required).
+        annotation_type can be 'none', 'image_bbox' (Required).
         """
         endpoint = "datasets"
         payload = {