Proper fix of renaming

qdrant · Dec 18, 2024 · cedce3f · cedce3f
1 parent 3b46c80
commit cedce3f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tools/async_client_generator/transformers/remote/import_from_transformer.py b/tools/async_client_generator/transformers/remote/import_from_transformer.py
@@ -19,8 +19,7 @@ def visit_ImportFrom(self, node: ast.ImportFrom) -> ast.AST:
                 for old_value, new_value in self.import_replace_map.items():
                     alias.name = alias.name.replace(old_value, new_value)
                 if alias.name == "get_channel":
-                    alias.name = "get_async_channel as get_channel"
-                    # alias.asname = "get_channel"
-        # node.names = [alias for alias in node.names if alias.name != "get_channel"]
+                    alias.name = "get_async_channel"
+                    alias.asname = "get_channel"
 
         return self.generic_visit(node)