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)