Environment Installation Failure Due to Version Conflict

[YichengZou626]

Hi,

I attempted to replicate the experiment results but encountered an issue while trying to set up the environment. When I ran the following commands:

create the environment

conda env create -f environment.yml

activate the environment

conda activate pdediff

I received the following error message:

Pip subprocess output:
Collecting dawgz==0.4.1
Downloading dawgz-0.4.1-py3-none-any.whl (13 kB)
Collecting jax==0.4.4
Downloading jax-0.4.4.tar.gz (1.2 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.2/1.2 MB 14.7 MB/s eta 0:00:00
Preparing metadata (setup.py): started
Preparing metadata (setup.py): finished with status 'done'

Pip subprocess error:
ERROR: Ignored the following versions that require a different python version:
0.4.31 Requires-Python >=3.10; 0.4.32 Requires-Python >=3.10; 0.4.33 Requires-Python >=3.10; 0.4.34 Requires-Python >=3.10; 0.4.35 Requires-Python >=3.10

ERROR: Could not find a version that satisfies the requirement jaxlib==0.4.4
(from versions: 0.4.6, 0.4.7, 0.4.9, 0.4.10, 0.4.11, 0.4.12, 0.4.13, 0.4.14,
0.4.16, 0.4.17, 0.4.18, 0.4.19, 0.4.20, 0.4.21, 0.4.22, 0.4.23, 0.4.24, 0.4.25,
0.4.26, 0.4.27, 0.4.28, 0.4.29, 0.4.30)

ERROR: No matching distribution found for jaxlib==0.4.4

failed

CondaEnvException: Pip failed

Based on the error, it seems there might be a version conflict between jaxlib and the Python version specified in the environment. Could you confirm the Python version intended for this environment? If the issue is indeed caused by a version mismatch, updating the environment.yml file with compatible versions would be greatly appreciated.

Thank you for your help!

[federicobergamin]

Hi Yicheng,

thanks for showing interest in the codebase and also thanks for pointing this out. You are correct, there is something wrong with the environment. It might be mostly because they have removed jaxlib==0.4.4 from pip installation https://pypi.org/project/jaxlib/#history. As a workaround I managed to install the environment locally by changing the Jax and jaxlib version to 0.4.30, i.e. in the environment.yml file you should use

 - jax==0.4.30
 - jaxlib==0.4.30

instead of

 - jax==0.4.4
 - jaxlib==0.4.4

Note that our method is never using JAX. You need JAX just to generate the Kolmogorov flow dataset, so I would assume that you should be able to run everything also with a different version of JAX. We will update the repo soon then to point this thing out.

Also I apologize for the late reply. Hope my answer will help, otherwise feel free to follow up.