From 3e0fcdd5c555146138a4034c1737eccfef3a6797 Mon Sep 17 00:00:00 2001 From: Yufeng Shi Date: Fri, 28 Feb 2025 14:17:16 +0000 Subject: [PATCH] Fix docstring of to_backend in backend_api.py The second method overloading of to_backend() has the wrong signature in docstring Change-Id: I9e1c5eccea00003b5e2821e666a9e0f77902d794 Signed-off-by: Yufeng Shi --- exir/backend/backend_api.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/exir/backend/backend_api.py b/exir/backend/backend_api.py index 966cae5f022..519f184871a 100644 --- a/exir/backend/backend_api.py +++ b/exir/backend/backend_api.py @@ -1,5 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. +# Copyright 2025 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -56,9 +57,9 @@ def to_backend( ) -> LoweredBackendModule: def to_backend( - graph_module: torch.fx.GraphModule, - partitioner: Type[TPartitioner], - ) -> torch.fx.GraphModule + edge_program: ExportedProgram, + partitioner: Partitioner, + ) -> ExportedProgram: """ pass