Skip to content

Commit 88e4e0b

Browse files
tcloseeffigies
authored andcommitted
adds bytes repr implementation for mock fileset types to get pydra task doctests to work
1 parent 00b92ec commit 88e4e0b

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

pydra/utils/hash.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
)
2121
from filelock import SoftFileLock
2222
import attrs.exceptions
23-
from fileformats.core import FileSet
23+
from fileformats.core.fileset import FileSet, MockMixin
2424
from . import user_cache_dir, add_exc_note
2525

2626
logger = logging.getLogger("pydra")
@@ -485,6 +485,13 @@ def bytes_repr_fileset(
485485
yield from fileset.__bytes_repr__(cache)
486486

487487

488+
@register_serializer(MockMixin)
489+
def bytes_repr_mock_fileset(
490+
mock_fileset: MockMixin, cache: Cache
491+
) -> Iterator[ty.Union[CacheKey, bytes]]:
492+
yield from mock_fileset.__bytes_repr__(cache)
493+
494+
488495
@register_serializer(list)
489496
@register_serializer(tuple)
490497
def bytes_repr_seq(obj: Sequence, cache: Cache) -> Iterator[bytes]:

0 commit comments

Comments
 (0)