Limit the upper versions of DALI wheel installation dependencies (#5710)

* Limit the maximum version of nvimagecodec for current DALI * Limit the upper versions of DALI wheel installation dependencies --------- Signed-off-by: Joaquin Anton Guirao <[email protected]> Signed-off-by: Kamil Tokarski <[email protected]> Co-authored-by: Joaquin Anton Guirao <[email protected]>
NVIDIA · Nov 13, 2024 · e0ca4e4 · e0ca4e4
1 parent b8568ca
commit e0ca4e4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/dali/python/setup.py.in b/dali/python/setup.py.in
@@ -82,13 +82,13 @@ For more details please check the
       install_requires=[
           # gast and astunparse are required for AutoGraph.
           # Currently supported range of versions.
-          'astunparse >= 1.6.0',
-          'gast >= 0.3.3',
+          'astunparse >= 1.6.0, <= 1.6.3',
+          'gast >= 0.3.3, <= 0.6.0',
           # the latest astunparse (1.6.3) doesn't work with any other six than
           # 1.16 on python 3.12 due to import six.moves
-          'six >= 1.16',
-          'dm-tree',
-          'packaging',
+          'six >= 1.16, <= 1.16',
+          'dm-tree <= 0.1.8',
+          'packaging <= 24.2',
           @DALI_INSTALL_REQUIRES_NVIMGCODEC@
           ],
      )