Pytorch dependency for Flair too restrictive

[indylec]

Hi,

When trying to pip install polyfuzz[flair], pip is not able to resolve the pytorch dependency:

INFO: pip is looking at multiple versions of polyfuzz[flair] to determine which version is compatible with other requirements. This could take a while.
Collecting polyfuzz[flair]
  Downloading polyfuzz-0.2.0-py2.py3-none-any.whl (25 kB)
  Downloading polyfuzz-0.0.1-py2.py3-none-any.whl (21 kB)
WARNING: polyfuzz 0.0.1 does not provide the extra 'flair'
Collecting sparse-dot-topn>=0.2.9
  Downloading sparse_dot_topn-0.3.1.tar.gz (17 kB)
  Installing build dependencies ... done
  Getting requirements to build wheel ... done
  Installing backend dependencies ... done
  Preparing metadata (pyproject.toml) ... done
ERROR: Cannot install polyfuzz[flair]==0.2.2, polyfuzz[flair]==0.3.0, polyfuzz[flair]==0.3.1, polyfuzz[flair]==0.3.2, polyfuzz[flair]==0.3.3, polyfuzz[flair]==0.3.4 and polyfuzz[flair]==0.4.0 because these package versions have conflicting dependencies.

The conflict is caused by:
    polyfuzz[flair] 0.4.0 depends on torch<1.7.1 and >=1.4.0; extra == "flair"
    polyfuzz[flair] 0.3.4 depends on torch<1.7.1 and >=1.4.0; extra == "flair"
    polyfuzz[flair] 0.3.3 depends on torch<1.7.1 and >=1.4.0; extra == "flair"
    polyfuzz[flair] 0.3.2 depends on torch>=1.2.0; extra == "flair"
    polyfuzz[flair] 0.3.1 depends on torch>=1.2.0; extra == "flair"
    polyfuzz[flair] 0.3.0 depends on torch>=1.2.0; extra == "flair"
    polyfuzz[flair] 0.2.2 depends on torch>=1.2.0; extra == "flair"

To fix this you could try to:
1. loosen the range of package versions you've specified
2. remove package versions to allow pip attempt to solve the dependency conflict

Neither pip nor conda offers pytorch version less than 1.9, current release is 1.13 - is the strict dependency something that can be updated without breaking anything?

Or have I just missed something obvious here?

Standard polyfuzz 0.4 install works just fine, otherwise - but I'd like to try out flair!

Thanks, Indy