phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 23:29:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Xi Yan 2024-10-23 13:07:15 -07:00
parent 92e32f80ad
commit 555f6e1531
2 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
View file

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

2
llama_stack/providers/registry/inference.py
View file

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