Merge pull request #177 from fastlabel/feature/dataset-license

add: license args in create_dataset method

Author: Jiei-S

README.md

@@ -2393,6 +2393,7 @@ Create a new dataset.
 ```python
 dataset = client.create_dataset(
     name="object-detection", # Only lowercase alphanumeric characters + hyphen is available
+    license="The MIT License" # Optional
 )
 ```
 
@@ -2404,6 +2405,7 @@ See API docs for details.
 {
     'id': 'YOUR_DATASET_ID',
     'name': 'object-detection',
+    'license': 'The MIT License',
     'createdAt': '2022-10-31T02:20:00.248Z',
     'updatedAt': '2022-10-31T02:20:00.248Z'
 }

examples/create_dataset.py

@@ -4,5 +4,5 @@
 
 client = fastlabel.Client()
 
-dataset = client.create_dataset(name="object-detection")
+dataset = client.create_dataset(name="object-detection", license="The MIT License")
 pprint(dataset)

fastlabel/__init__.py

@@ -3860,19 +3860,15 @@ def get_datasets(
             params["limit"] = limit
         return self.api.get_request(endpoint, params=params)
 
-    def create_dataset(
-        self,
-        name: str,
-    ) -> dict:
+    def create_dataset(self, name: str, license: str = "") -> dict:
         """
         Create a dataset.
 
         name is name of your dataset. Only lowercase alphanumeric characters + hyphen is available (Required).
+        license is a license name of your dataset. (Optional)
         """
         endpoint = "datasets"
-        payload = {
-            "name": name,
-        }
+        payload = {"name": name, "license": license}
         return self.api.post_request(endpoint, payload=payload)
 
     def update_dataset(