update requirements

requirements.txt

@@ -1,24 +1,45 @@
+absl-py==2.3.1
 aiohappyeyeballs==2.6.1
 aiohttp==3.12.15
 aiosignal==1.4.0
+antlr4-python3-runtime==4.9.3
 attrs==25.3.0
+bitsandbytes==0.48.0
+black @ git+https://github.com/psf/black@4a8e6b789bbd18c0455d74bbec8d957e824eb3bf
 certifi==2025.8.3
 charset-normalizer==3.4.3
+click==8.3.0
+contourpy==1.3.3
+cycler==0.12.1
 datasets==4.1.1
 dill==0.4.0
+docstring_parser==0.17.0
 filelock==3.19.1
+fonttools==4.60.0
 frozenlist==1.7.0
 fsspec==2025.9.0
+grpcio==1.75.0
 hf-xet==1.1.10
 huggingface-hub==0.35.0
+hydra-core==1.3.2
 idna==3.10
+importlib_resources==6.5.2
+iniconfig==2.1.0
 Jinja2==3.1.6
+jsonargparse==4.41.0
+jsonnet==0.21.0
+kiwisolver==1.4.9
 lightning==2.5.5
 lightning-utilities==0.15.2
+Markdown==3.9
+markdown-it-py==4.0.0
 MarkupSafe==3.0.2
+matplotlib==3.10.6
+mdurl==0.1.2
 mpmath==1.3.0
 multidict==6.6.4
 multiprocess==0.70.16
+mypy_extensions==1.1.0
 networkx==3.5
 numpy==2.3.3
 nvidia-cublas-cu12==12.8.4.1
@@ -35,28 +56,44 @@ nvidia-cusparselt-cu12==0.7.1
 nvidia-nccl-cu12==2.27.3
 nvidia-nvjitlink-cu12==12.8.93
 nvidia-nvtx-cu12==12.8.90
+omegaconf==2.3.0
 packaging==25.0
 pandas==2.3.2
+pathspec==0.12.1
+pillow==11.3.0
+-e git+ssh://git@github.com/mathLab/PINA@b5e4d13663d1e9f890472b3e000583b57cc45d17#egg=pina_mathlab
+platformdirs==4.4.0
+pluggy==1.6.0
 propcache==0.3.2
+protobuf==6.32.1
 psutil==7.1.0
 pyarrow==21.0.0
+Pygments==2.19.2
 pyparsing==3.2.5
+pytest==8.4.2
 python-dateutil==2.9.0.post0
+pytokens==0.1.10
 pytorch-lightning==2.5.5
 pytz==2025.2
 PyYAML==6.0.2
 requests==2.32.5
+rich==14.1.0
 setuptools==78.1.1
 six==1.17.0
 sympy==1.14.0
+tensorboard==2.20.0
+tensorboard-data-server==0.7.2
+tensorboardX==2.6.4
 torch==2.8.0
 torch-geometric==2.6.1
 torchmetrics==1.8.2
 tqdm==4.67.1
 triton==3.4.0
+typeshed_client==2.8.2
 typing_extensions==4.15.0
 tzdata==2025.2
 urllib3==2.5.0
+Werkzeug==3.1.3
 wheel==0.45.1
 xxhash==3.5.0
 yarl==1.20.1