def test_metadata_save(self): filename = "dummy_target.md" filepath = Path(REPOCARD_DIR) / filename filepath.write_text(DUMMY_MODELCARD) metadata_save(filepath, {"meaning_of_life": 42}) content = filepath.read_text() self.assertEqual(content, DUMMY_MODELCARD_TARGET)
def test_metadata_save_from_file_no_yaml(self): filename = "dummy_target_2.md" filepath = Path(REPOCARD_DIR) / filename filepath.write_text("Hello\n") metadata_save(filepath, {"meaning_of_life": 42}) content = filepath.read_text() self.assertEqual(content, DUMMY_MODELCARD_TARGET_NO_YAML)
def test_metadata_eval_result(self): data = metadata_eval_result( model_pretty_name="RoBERTa fine-tuned on ReactionGIF", task_pretty_name="Text Classification", task_id="text-classification", metrics_pretty_name="Accuracy", metrics_id="accuracy", metrics_value=0.2662102282047272, dataset_pretty_name="ReactionGIF", dataset_id="julien-c/reactiongif", ) filename = "eval_results.md" filepath = Path(REPOCARD_DIR) / filename metadata_save(filepath, data) content = filepath.read_text().splitlines() self.assertEqual(content, DUMMY_MODELCARD_EVAL_RESULT.splitlines())
def repocard_metadata_save(self, data: Dict) -> None: return metadata_save(os.path.join(self.local_dir, REPOCARD_NAME), data)