"StrEnum" import may be missing.

[jianhan-amd]

In some of the container enum and backports.strenum may not be pre-installed. It is imported in "TraceLens/TreePerf/jax_analyses.py". Shall we add the package enum or backports.strenum into the installation script?

try:
    from enum import StrEnum
except ImportError:
    from backports.strenum import StrEnum

@jasainio @gabeweisz

[jakki-amd]

StrEnum was introduced in Python 3.11, setup.py states python_requires='>=3.6', so this could be updated to 3.11. Although I went through the source code where StrEnum was used and I did not see case where normal Enum class could not be used.

[vorapolsiloai]

can confirm that I have to do pip install backports.strenum

to get rid of

(.tracelens) vassavas@amd.com@tw046:~/projects/TraceLens$ python3 examples/generate_perf_report.py --help
Traceback (most recent call last):
  File "/home/vassavas@amd.com/projects/TraceLens/.tracelens/lib/python3.10/site-packages/TraceLens/TreePerf/jax_analyses.py", line 9, in <module>
    from enum import StrEnum
ImportError: cannot import name 'StrEnum' from 'enum' (/usr/lib/python3.10/enum.py)

[jianhan-amd]

StrEnum was introduced in Python 3.11, setup.py states python_requires='>=3.6', so this could be updated to 3.11. Although I went through the source code where StrEnum was used and I did not see case where normal Enum class could not be used.

I am not sure if updating python_requires='>=3.6 -> python_requires='>=3.11 is a good idea. We have environments with Python3.10 and sometime we want to integrate Tracelens working into these environments.

[gabeweisz]

StrEnum works with Python 3.10 if you install backports.strenum. Is there a way to tell pip to install backports.strenum if python <3.11?

[jakki-amd]

Based on this documentation, you can define environment conditional requirements in setup.py with:

install_requires=[
        'pandas',
        'backports.strenum;python_version<"3.11"'
],

[gabeweisz]

Closing this as I have created the PR