fix requiriments

activeloopai · Sep 6, 2024 · 9454e53 · 9454e53
1 parent 3ca5260
commit 9454e53
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/deeplake/core/compression.py b/deeplake/core/compression.py
@@ -1194,7 +1194,7 @@ def _open_3d_data(file):
     return point_cloud
 
 
-def _open_mesh_data(file):
+def _open_mesh_data(file: Union[bytes, memoryview, str]):
     from stl import mesh
 
     if isinstance(file, str):
@@ -1213,12 +1213,12 @@ def _read_3d_data_shape_and_dtype(file: Union[bytes, BinaryIO]):
     return point_cloud.shape, point_cloud.dtype
 
 
-def _read_stl_shape_and_dtype(file: Union[bytes, memoryview, str]):
+def _read_stl_shape_and_dtype(file):
     mesh_data = _open_mesh_data(file)
     return mesh_data.vectors.shape, mesh_data.vectors.dtype
 
 
-def _decompress_stl(file: Union[bytes, memoryview, str]):
+def _decompress_stl(file: Union[bytes, str]):
     mesh_data = _open_mesh_data(file)
     return mesh_data.vectors
 

diff --git a/deeplake/requirements/common.txt b/deeplake/requirements/common.txt
@@ -11,7 +11,7 @@ pathos
 humbug>=0.3.1
 tqdm
 lz4
-av>=8.1.0; python_version >= '3.7' or sys_platform != 'win32'
+av>=8.1.0,<=12.3.0; python_version >= '3.7' or sys_platform != 'win32'
 pydicom
 IPython
 flask

diff --git a/deeplake/requirements/plugins.txt b/deeplake/requirements/plugins.txt
@@ -10,4 +10,4 @@ mmdet==2.28.1; platform_system == "Linux" and python_version >= "3.7"
 mmsegmentation==0.30.0; platform_system == "Linux" and python_version >= "3.7"
 mmengine
 pandas
-av
+av==12.3.0