Merge pull request #14 from steven-murray/yaml-error

fix: the yaml loader

src/hickleable/__init__.py

@@ -244,8 +244,8 @@ def _load_path(h_node, base_type, py_obj_type):
 
 if yaml is not None:
     # Register a YAML loader for hickle-dumped files.
-    def _hickle_yaml_loader(path):
-        return hickle.load(path)
+    def _hickle_yaml_loader(loader: yaml.SafeLoader, node):
+        return hickle.load(node.value)
 
     yaml.add_constructor("!hickle", _hickle_yaml_loader, Loader=yaml.FullLoader)
     yaml.add_constructor("!hickle", _hickle_yaml_loader, Loader=yaml.Loader)

tests/test_yaml.py

@@ -0,0 +1,25 @@
+import hickle
+import yaml
+
+from hickleable import hickleable
+
+
+@hickleable()
+class A:
+    def __init__(self, a):
+        self.a = a
+
+    def __eq__(self, other):
+        return self.a == other.a
+
+
+def test_load_yaml(tmpdir):
+    fl = tmpdir / "test_load_yaml.h5"
+    a = A("more stuff")
+
+    hickle.dump(a, str(fl))
+
+    txt = f"""!hickle {fl}"""
+
+    b = yaml.load(txt, Loader=yaml.FullLoader)
+    assert a == b