Spaces:

TRI-ML
/

risk_biased_prediction

Running

jmercat commited on Jul 1

Commit

6865b66

•

1 Parent(s): 9661983

trust_remote_code=true when loading private dataset

Files changed (1) hide show

scripts/scripts_utils/plotly_interface.py CHANGED Viewed

@@ -345,7 +345,7 @@ def load_predictor_from_hf(model_source: str = "TRI-ML/risk_biased_model", confi
     return predictor
 def load_dataset_from_hf(data_source: str = "jmercat/risk_biased_dataset") -> Dataset:
-    dataset = load_dataset(data_source, split="test")
     return dataset
 def main(load_from=None, cfg_path=None):

     return predictor
 def load_dataset_from_hf(data_source: str = "jmercat/risk_biased_dataset") -> Dataset:
+    dataset = load_dataset(data_source, split="test", trust_remote_code=True)
     return dataset
 def main(load_from=None, cfg_path=None):