Added more detailed logging.

Author: adivekar-utexas

src/fmcore/framework/_dataset.py

@@ -85,6 +85,10 @@ def concat(
     @model_validator(mode="before")
     @classmethod
     def _set_dataset_params(cls, params: Dict) -> Dict:
+        if "data_schema" not in params:
+            raise ValueError(
+                f"Cannot create instance of class '{cls.class_name}' without passing `data_schema` parameter."
+            )
         data_schema: Union[Schema, MLTypeSchema] = params["data_schema"]
         if isinstance(data_schema, dict):
             ## We need to infer the schema:

src/fmcore/framework/_predictions.py

@@ -36,8 +36,8 @@
     MLTypeSchema,
     TaskOrStr,
 )
-from fmcore.framework._task_mixins import InputOutputDataMixin, SchemaValidationError
 from fmcore.framework._dataset import Dataset
+from fmcore.framework._task_mixins import InputOutputDataMixin, SchemaValidationError
 
 Predictions = "Predictions"
 Visualization = "Visualization"
@@ -69,6 +69,11 @@ def _pre_registration_hook(cls):
     @model_validator(mode="before")
     @classmethod
     def _set_predictions_params(cls, params: Dict) -> Dict:
+        if "data_schema" not in params:
+            raise ValueError(
+                f"Cannot create instance of class '{cls.class_name}' without passing `data_schema` parameter."
+            )
+
         params["data_schema"]: Schema = Schema.of(params["data_schema"], schema_template=cls.schema_template)
         # data_schema: Union[Schema, MLTypeSchema] = params['data_schema']
         # if isinstance(data_schema, dict):