diff --git a/environment.yml b/environment.yml index e2a5275b4..fc7d86422 100644 --- a/environment.yml +++ b/environment.yml @@ -4,3 +4,6 @@ channels: dependencies: - numpy<2.0 - pytorch>=1.10 + - matplotlib-base + - tqdm + - packaging