propel comment

Author: alanzhang25

src/judgeval/dataset/__init__.py

@@ -153,21 +153,17 @@ def create(
         )
         judgeval_logger.info(f"Created dataset {name}")
 
-        dataset_instance = cls(
-            name=name,
-            project_name=project_name,
-            examples=[],
-        )
-        dataset_instance.add_examples(examples, batch_size=batch_size)
-
         if not isinstance(examples, list):
             examples = list(examples)
 
-        return cls(
+        dataset = cls(
             name=name,
             project_name=project_name,
             examples=examples,
         )
+        dataset.add_examples(examples, batch_size=batch_size)
+
+        return dataset
 
     @classmethod
     def list(cls, project_name: str):

src/judgeval/v1/datasets/dataset_factory.py

@@ -77,19 +77,14 @@ def create(
         )
         judgeval_logger.info(f"Created dataset {name}")
 
-        dataset = Dataset(
-            name=name, project_name=project_name, examples=[], client=self._client
-        )
-        dataset.add_examples(examples, batch_size=batch_size)
         if not isinstance(examples, list):
             examples = list(examples)
 
-        return Dataset(
-            name=name,
-            project_name=project_name,
-            examples=examples,
-            client=self._client,
+        dataset = Dataset(
+            name=name, project_name=project_name, examples=examples, client=self._client
         )
+        dataset.add_examples(examples, batch_size=batch_size)
+        return dataset
 
     def list(self, project_name: str) -> List[DatasetInfo]:
         datasets = self._client.datasets_pull_all_for_judgeval(