File size: 781 Bytes
6980f6d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import typing
try:
from typing import Self
except ImportError:
from typing_extensions import Self
typing.Self = Self
import enum
if not hasattr(enum, "StrEnum"):
class StrEnum(str, enum.Enum):
def __str__(self):
return str(self.value)
enum.StrEnum = StrEnum
import sys
from litert_torch.generative.export_hf.export_main import main
if __name__ == "__main__":
# If main expects an argument like in absl.app.run(main), it usually passes argv.
# But since it failed with missing 1 positional argument '_', let's try passing sys.argv.
try:
main(sys.argv)
except Exception as e:
print(f"Failed with sys.argv: {e}")
# Try fire-like or direct call if argv didn't work
# (Though usually it's absl)
|