paperswithcode / torchbench

Easily benchmark machine learning models in PyTorch
Apache License 2.0
147 stars 20 forks source link

Suggest to loosen the dependency on albumentations #20

Open Agnes-U opened 2 years ago

Agnes-U commented 2 years ago

Hi, your project torchbench(commit id: dca5f740ff58b2e419db3ca1eb38e1cd3cb658f4) requires "albumentations==0.3.0" in its dependency. After analyzing the source code, we found that the following versions of albumentations can also be suitable, i.e., albumentations 0.1.8, 0.1.9, 0.1.10, 0.1.11, 0.1.12, 0.2.0, 0.2.1, 0.2.2, 0.2.3, since all functions that you directly (1 APIs: albumentations.core.transforms_interface.DualTransform.init) or indirectly (propagate to 1 albumentations's internal APIs and 0 outsider APIs) used from the package have not been changed in these versions, thus not affecting your usage.

Therefore, we believe that it is quite safe to loose your dependency on albumentations from "albumentations==0.3.0" to "albumentations>=0.1.8,<=0.3.0". This will improve the applicability of torchbench and reduce the possibility of any further dependency conflict with other projects.

May I pull a request to further loosen the dependency on albumentations?

By the way, could you please tell us whether such an automatic tool for dependency analysis may be potentially helpful for maintaining dependencies easier during your development?