From 590bfb5770d9e9430c92ac90c0112fcbaebb1a7f Mon Sep 17 00:00:00 2001 From: Adrien Vannson Date: Fri, 20 Dec 2024 18:00:24 +0100 Subject: [PATCH] Fix import --- src/betterproto2_compiler/compile/importing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/betterproto2_compiler/compile/importing.py b/src/betterproto2_compiler/compile/importing.py index 82bd0a21..814fff71 100644 --- a/src/betterproto2_compiler/compile/importing.py +++ b/src/betterproto2_compiler/compile/importing.py @@ -105,9 +105,9 @@ def get_type_reference( compiling_google_protobuf = current_package == ["google", "protobuf"] importing_google_protobuf = py_package == ["google", "protobuf"] if importing_google_protobuf and not compiling_google_protobuf: - py_package = ["betterproto", "lib"] + (["pydantic"] if pydantic else []) + py_package + py_package = ["betterproto2", "lib"] + (["pydantic"] if pydantic else []) + py_package - if py_package[:1] == ["betterproto"]: + if py_package[:1] == ["betterproto2"]: return reference_absolute(imports, py_package, py_type) if py_package == current_package: