style(executors): replace double quotes, add type hint

jina-ai · May 17, 2020 · 7aceae3 · 7aceae3
1 parent 4d11e11
commit 7aceae3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/jina/executors/crafters/numeric/io.py b/jina/executors/crafters/numeric/io.py
@@ -8,7 +8,7 @@ class ArrayReader(BaseSegmenter):
         Numbers are split on the provided delimiter, default is comma (,)
     """
 
-    def __init__(self, delimiter=",", as_type="float32", *args, **kwargs):
+    def __init__(self, delimiter: str = ',', as_type: str = 'float32', *args, **kwargs):
         """
         :param delimiter: delimiter between numbers
         :param as_type: type of number

diff --git a/tests/executors/crafters/numeric/test_io.py b/tests/executors/crafters/numeric/test_io.py
@@ -8,13 +8,13 @@
 class MyTestCase(JinaTestCase):
     def test_array_reader(self):
         size = 8
-        sample_array = np.random.rand(size).astype("float32")
-        raw_bytes = ",".join([str(x) for x in sample_array]).encode("utf8")
+        sample_array = np.random.rand(size).astype('float32')
+        raw_bytes = ','.join([str(x) for x in sample_array]).encode('utf8')
 
         reader = ArrayReader()
         crafted_chunk = reader.craft(raw_bytes, 0)[0]
 
-        np.testing.assert_array_equal(crafted_chunk["blob"], sample_array)
+        np.testing.assert_array_equal(crafted_chunk['blob'], sample_array)
 
 if __name__ == '__main__':
     unittest.main()