diff --git a/flytekit/remote/remote.py b/flytekit/remote/remote.py index 09c44f949b..fb47b99a40 100644 --- a/flytekit/remote/remote.py +++ b/flytekit/remote/remote.py @@ -358,8 +358,14 @@ def get( data_response = self.client.get_data(flyte_uri) if data_response.HasField("literal_map"): + from flytekit.models.interface import VariableMap + lm = LiteralMap.from_flyte_idl(data_response.literal_map) - return LiteralsResolver(lm.literals) + if data_response.variable_map is not None: + vm = VariableMap.from_flyte_idl(data_response.variable_map) + return LiteralsResolver(lm.literals, vm.variables) + else: + return LiteralsResolver(lm.literals) elif data_response.HasField("literal"): return Literal.from_flyte_idl(data_response.literal) elif data_response.HasField("pre_signed_urls"):