fix callable type

Author: lucidrains

perfusion_pytorch/embedding.py

@@ -59,7 +59,7 @@ def __init__(
         num_concepts = 1,
         superclass_embed_id: Optional[Union[int, Tuple[int, ...]]] = None,
         superclass_string: Optional[str] = None,
-        tokenize: Callable[str, Tensor] = tokenizer.tokenize,
+        tokenize: Callable[[List[str]], Tensor] = tokenizer.tokenize,
         tokenizer_pad_id: int = 0,
         tokenizer_sos_eos_id: Tuple[int, int] = (49406, 49407)
     ):

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'perfusion-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.1.12',
+  version = '0.1.14',
   license='MIT',
   description = 'Perfusion - Pytorch',
   author = 'Phil Wang',