Fixing the lint error.

Signed-off-by: lrangine <[email protected]>
feast-dev · Dec 14, 2024 · eda6900 · eda6900
1 parent cc40246
commit eda6900
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sdk/python/feast/infra/registry/remote.py b/sdk/python/feast/infra/registry/remote.py
@@ -77,7 +77,7 @@ def __init__(
         assert isinstance(registry_config, RemoteRegistryConfig)
         # self.channel = create_tls_channel(registry_config)
 
-        self._create_grpc_channel(registry_config)
+        self.channel =   self._create_grpc_channel(registry_config)
 
         auth_header_interceptor = GrpcClientAuthHeaderInterceptor(auth_config)
         self.channel = grpc.intercept_channel(self.channel, auth_header_interceptor)
@@ -99,10 +99,10 @@ def _create_grpc_channel(self, registry_config):
             tls_credentials = grpc.ssl_channel_credentials(
                 root_certificates=trusted_certs
             )
-            self.channel = grpc.secure_channel(registry_config.path, tls_credentials)
+            return grpc.secure_channel(registry_config.path, tls_credentials)
         else:
             # Create an insecure gRPC channel
-            self.channel = grpc.insecure_channel(registry_config.path)
+            return grpc.insecure_channel(registry_config.path)
 
     def close(self):
         if self.channel: