fix: type for md5_bytes

Author: rememberYou

pyrdf2vec/walkers/community.py

@@ -2,7 +2,7 @@
 import math
 from collections import defaultdict
 from hashlib import md5
-from typing import List, Set
+from typing import List, Optional, Set
 
 import attr
 import community
@@ -76,7 +76,13 @@ class CommunityWalker(Walker):
         type=float,
         validator=attr.validators.instance_of(float),
     )
-    md5_bytes = attr.ib(kw_only=True, default=8, type=int, repr=False)
+
+    md5_bytes = attr.ib(
+        kw_only=True,
+        type=Optional[int],
+        default=8,
+        repr=False,
+    )
 
     resolution = attr.ib(
         kw_only=True,

pyrdf2vec/walkers/halk.py

@@ -54,8 +54,6 @@ class HALKWalker(RandomWalker):
         ),
     )
 
-    md5_bytes = attr.ib(kw_only=True, default=8, type=int, repr=False)
-
     def build_dictionary(
         self, walks: List[SWalk]
     ) -> DefaultDict[str, Set[int]]:

pyrdf2vec/walkers/random.py

@@ -1,5 +1,5 @@
 from hashlib import md5
-from typing import List, Set
+from typing import List, Optional, Set
 
 import attr
 
@@ -39,7 +39,12 @@ class RandomWalker(Walker):
 
     """
 
-    md5_bytes = attr.ib(kw_only=True, default=8, type=int, repr=False)
+    md5_bytes = attr.ib(
+        kw_only=True,
+        type=Optional[int],
+        default=8,
+        repr=False,
+    )
 
     def _bfs(
         self, kg: KG, entity: Vertex, is_reverse: bool = False

pyrdf2vec/walkers/weisfeiler_lehman.py

@@ -46,8 +46,6 @@ class WLWalker(RandomWalker):
 
     """
 
-    md5_bytes = attr.ib(kw_only=True, default=8, type=int, repr=False)
-
     wl_iterations = attr.ib(
         kw_only=True,
         default=4,