Skip to content

Commit

Permalink
Skip paddle version when no version is provided
Browse files Browse the repository at this point in the history
Signed-off-by: Szymon Karpiński <skarpinski@nvidia.com>
  • Loading branch information
szkarpinski committed Jan 16, 2024
1 parent f17a5f8 commit 07b558a
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions dali/python/nvidia/dali/plugin/paddle.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
# Copyright (c) 2017-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# Copyright (c) 2017-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -25,10 +25,12 @@
from nvidia.dali.plugin.base_iterator import _DaliBaseIterator
from nvidia.dali.plugin.base_iterator import LastBatchPolicy


assert LooseVersion(paddle.__version__) == LooseVersion("0.0.0") or LooseVersion(
paddle.__version__
) >= LooseVersion("2.0.0"), "DALI PaddlePaddle support requires Paddle develop or release >= 2.0.0"
if isinstance(paddle.__version__, str):
assert LooseVersion(paddle.__version__) == LooseVersion("0.0.0") or LooseVersion(
paddle.__version__
) >= LooseVersion(
"2.0.0"
), "DALI PaddlePaddle support requires Paddle develop or release >= 2.0.0"


dtype_map = {
Expand Down

0 comments on commit 07b558a

Please sign in to comment.