diff --git a/pyproject.toml b/pyproject.toml index 7123b0750..ffe50d244 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta" [project] name = "chakra" -requires-python = ">=3.7" +requires-python = ">=3.10" version = "1.0.0" readme = "README.md" license = {file = "LICENSE.md"} @@ -49,7 +49,7 @@ chakra_trace_link = "chakra.src.trace_link.trace_link:main" chakra_visualizer = "chakra.src.visualizer.visualizer:main" [tool.ruff] -target-version = "py39" +target-version = "py310" line-length = 120 exclude = [ "src/converter/text_converter.py", diff --git a/setup.py b/setup.py index cf2036f1c..1be724b2b 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,7 @@ from setuptools import setup from setuptools.command.build import build + class build_grpc(build): """ Custom build class to include gRPC build commands. diff --git a/src/converter/pytorch_node.py b/src/converter/pytorch_node.py index cb7bab703..f366421db 100644 --- a/src/converter/pytorch_node.py +++ b/src/converter/pytorch_node.py @@ -47,7 +47,13 @@ class PyTorchNode: pg_name (str): Process Group name for the inter-GPU communication. """ - SUPPORTED_VERSIONS = ["1.0.2-chakra.0.0.4", "1.0.3-chakra.0.0.4", "1.1.0-chakra.0.0.4", "1.1.1-chakra.0.0.4", "1.1.1-chakra.1.0.0"] + SUPPORTED_VERSIONS = [ + "1.0.2-chakra.0.0.4", + "1.0.3-chakra.0.0.4", + "1.1.0-chakra.0.0.4", + "1.1.1-chakra.0.0.4", + "1.1.1-chakra.1.0.0" + ] def __init__(self, schema: str, node_data: Dict[str, Any]) -> None: """ @@ -218,7 +224,7 @@ def comm_size(self) -> int: int: The calculated communication size. """ comm_size = 0 - for input_value, input_type in zip(self.inputs["values"], self.inputs["types"]): + for input_value, input_type in zip(self.inputs["values"], self.inputs["types"], strict=False): if "Tensor" in input_type: if input_type.startswith("GenericList[Tensor"): for inner_value in input_value: