Update tests/test_repocard.py

Author: Wauplin

src/huggingface_hub/repocard.py

@@ -331,7 +331,7 @@ def from_template(
         if template_str is None:
             template_str = Path(cls.default_template_path).read_text()
         template = jinja2.Template(template_str)
-        content = template.render(card_data=card_data.to_yaml(original_order=cls._original_order), **kwargs)
+        content = template.render(card_data=card_data.to_yaml(), **kwargs)
         return cls(content)
 
 

src/huggingface_hub/repocard_data.py

@@ -207,7 +207,9 @@ def to_yaml(self, line_break=None, original_order: Optional[List[str]] = None) -
         """
         if original_order:
             self.__dict__ = {
-                k: self.__dict__[k] for k in original_order + list(set(self.__dict__.keys()) - set(original_order))
+                k: self.__dict__[k]
+                for k in original_order + list(set(self.__dict__.keys()) - set(original_order))
+                if k in self.__dict__
             }
         return yaml_dump(self.to_dict(), sort_keys=False, line_break=line_break).strip()
 

tests/test_repocard.py

@@ -850,7 +850,6 @@ def test_model_card_from_template_eval_results(self):
     def test_preserve_order_load_save(self):
         model_card = ModelCard(DUMMY_MODELCARD)
         model_card.data.license = "test"
-        model_card.content
         self.assertEqual(model_card.content, "---\nlicense: test\ndatasets:\n- foo\n- bar\n---\n\nHello\n")