cleanup

llama_stack/providers/impls/meta_reference/datasetio/datasetio.py

@@ -69,8 +69,7 @@ class PandasDataframeDataset(BaseDataset):
         # check all columns in dataset schema are present
         assert len(self.df.columns) == len(self.dataset_def.dataset_schema)
 
-        # check all types match
-        print(self.df.dtypes)
+        # TODO: type checking against column types in dataset schema
 
     def load(self) -> None:
         if self.df is not None:

llama_stack/providers/registry/inference.py

@@ -36,7 +36,7 @@ def available_providers() -> List[ProviderSpec]:
             pip_packages=(
                 META_REFERENCE_DEPS
                 + [
-                    "fbgemm-gpu==1.0.0",
+                    "fbgemm-gpu==0.8.0",
                 ]
             ),
             module="llama_stack.providers.impls.meta_reference.inference",