From fb0c43c4b7b3bafb384ff7bc8a31a46e28996c83 Mon Sep 17 00:00:00 2001 From: Alex Rogozhnikov Date: Sun, 17 Sep 2023 22:58:19 -0700 Subject: [PATCH 1/2] fix #279, update description --- einops/_backends.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/einops/_backends.py b/einops/_backends.py index ea85db04..0468722b 100644 --- a/einops/_backends.py +++ b/einops/_backends.py @@ -3,11 +3,11 @@ - backends are not imported unless those are actually needed, because - backends may not be installed - importing all available backends will drive to significant memory footprint - - backends may by present but installed with errors (but never used), + - backends may be present but installed with errors (but never used), importing may drive to crashes -- backend should be either symbolic or imperative (tensorflow is for both, but that causes problems) +- backend should be either symbolic or imperative - this determines which methods (from_numpy/to_numpy or create_symbol/eval_symbol) should be defined -- if backend can't (temporarily) provide symbols for shape dimensions, UnknownSize objects are used +- if backend can't provide symbols for shape dimensions, UnknownSize objects are used """ import sys @@ -29,7 +29,7 @@ def get_backend(tensor) -> "AbstractBackend": if _result is not None: return _result - for framework_name, backend in _loaded_backends.items(): + for framework_name, backend in list(_loaded_backends.items()): if backend.is_appropriate_type(tensor): _type2backend[_type] = backend return backend From fe9170607ea426de409649d8cf8c008e1639c828 Mon Sep 17 00:00:00 2001 From: Alex Rogozhnikov Date: Sun, 17 Sep 2023 23:06:38 -0700 Subject: [PATCH 2/2] use oneflow from pypi for tests --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index ba47b3e2..661210f0 100644 --- a/test.py +++ b/test.py @@ -41,7 +41,7 @@ def main(): "chainer": ["chainer"], "cupy": ["cupy"], "paddle": ["paddlepaddle==0.0.0 -f https://www.paddlepaddle.org.cn/whl/linux/cpu-mkl/develop.html"], - "oneflow": ["oneflow==0.9.0+cpu -f https://release.oneflow.info"], + "oneflow": ["oneflow==0.9.0"], } usage = f"""