# --------- pytorch --------- #
--find-links https://download.pytorch.org/whl/cu111/torch_stable.html
pytorch-lightning==1.7.7
torch==1.10.0+cu111
torchvision==0.11.0+cu111
torchmetrics==0.10.3
transformers
datasets==2.19.0

#torch>=1.8.1,<1.13.1
#torchvision==0.11.0

# --------- hydra --------- #
hydra-core==1.1.0
hydra-colorlog==1.1.0
hydra-optuna-sweeper==1.1.0
# hydra-ax-sweeper
# hydra-ray-launcher
# hydra-submitit-launcher
omegaconf

# --------- loggers --------- #
wandb
# neptune-client
# mlflow
# comet-ml
# torch_tb_profiler

# --------- linters --------- #
pre-commit      # hooks for applying linters on commit
black           # code formatting
isort           # import sorting
flake8          # code analysis

# --------- others --------- #
python-dotenv   # loading env variables from .env file
rich            # beautiful text formatting in terminal
pytest          # tests
sh              # for running bash commands in some tests
scikit-learn    # used in some callbacks
seaborn         # used in some callbacks
jupyterlab      # better jupyter notebooks
pudb            # debugger
setuptools==58.0.4 #Avoid bugs with tensorboad
gcsfs		# google cloud access tool
