feat: raise error in from_parquet when df is empty

opentargets · Nov 20, 2023 · a1c41e6 · a1c41e6
1 parent 1351728
commit a1c41e6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/otg/dataset/dataset.py b/src/otg/dataset/dataset.py
@@ -84,9 +84,14 @@ def from_parquet(
 
         Returns:
             Self: Dataset with the parquet file contents
+
+        Raises:
+            ValueError: Parquet file is empty
         """
         schema = cls.get_schema()
         df = session.read_parquet(path=path, schema=schema, **kwargs)
+        if df.isEmpty():
+            raise ValueError(f"Parquet file is empty: {path}")
         return cls(_df=df, _schema=schema)
 
     def validate_schema(self: Dataset) -> None: