Fix pyre in arm_backend.py

Differential Revision: D66475070 Pull Request resolved: pytorch#7069
kirklandsign · Nov 27, 2024 · 8b375f2 · 8b375f2
1 parent a2619e1
commit 8b375f2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backends/arm/arm_backend.py b/backends/arm/arm_backend.py
@@ -135,7 +135,9 @@ def set_quantize_io(self, quantize_io: bool = False) -> "ArmCompileSpecBuilder":
         self.quantize_io = quantize_io
         return self
 
-    def set_input_order(self, input_order: str = None) -> "ArmCompileSpecBuilder":
+    def set_input_order(
+        self, input_order: Optional[str] = None
+    ) -> "ArmCompileSpecBuilder":
         """
         Reorder the inputs coming in. This may be required when inputs > 1.
         And while using the U55/U85 CompileSpec.