Merge pull request #46 from amansrivastava17/ashutoshsingh0223-patch-1

Add dummy axis argument

server/embedding_as_service/utils.py

@@ -125,16 +125,16 @@ def download_from_url(url: str, download_path: str) -> None:
     sys.stdout.write('\n')
 
 
-def reduce_mean_max(vectors: np.ndarray):
+def reduce_mean_max(vectors: np.ndarray, axis=None):
     return np.hstack(np.mean(vectors, 0), np.max(vectors, 0))
 
 
-def np_first(vectors: np.ndarray):
-    return np.array(vectors)[0]
+def np_first(vectors: np.ndarray, axis=None):
+    return np.array(vectors)[:, 0, :]
 
 
-def np_last(vectors: np.ndarray):
-    return np.array(vectors)[-1]
+def np_last(vectors: np.ndarray, axis=None):
+    return np.array(vectors)[:, -1, :]
 
 
 POOL_FUNC_MAP = {
@@ -162,4 +162,4 @@ def __call__(cls, *args, **kwargs):
         argset = frozenset(sorted(argdict.items()))
         if argset not in cls._instanceDict:
             cls._instanceDict[argset] = super(ArgSingleton, cls).__call__(*args, **kwargs)
-        return cls._instanceDict[argset]
+        return cls._instanceDict[argset]