From 3b181cd13be1f5efd61bd67ca3c940bb468ae667 Mon Sep 17 00:00:00 2001
From: nyueki <67496212+nyueki@users.noreply.github.com>
Date: Wed, 5 Aug 2026 18:11:00 -0400
Subject: [PATCH] Refactor transport_dtype to transport_dtype_name

Updated transport_dtype to use transport_dtype_name for clarity.
---
 __init__.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 0c5b0ed..b6f86db 100644
--- a/__init__.py
+++ b/__init__.py
@@ -193,7 +193,12 @@ class RemoteCLIPProxy:
         self.ip = ip
         self.port = port
         self.transport_mode = transport_mode
-        self.transport_dtype = resolve_transport_dtype(transport_mode, ip)
+        self.transport_dtype_name = resolve_transport_dtype(transport_mode, ip)
+        self.transport_dtype = (
+            ALLOWED_DTYPES[self.transport_dtype_name]
+            if self.transport_dtype_name is not None
+            else None
+        )
         self._conn = connection or _Connection(ip, port, auth_token)
         self.lora_stack = list(lora_stack or [])
     def clone(self, disable_dynamic=False):
@@ -231,7 +236,7 @@ class RemoteCLIPProxy:
             "kwargs": kwargs,
             "lora_stack": tokens.get("lora_stack", self.lora_stack),
             "tensor_inputs": in_meta,
-            "transport_dtype": self.transport_dtype,
+            "transport_dtype": self.transport_dtype_name,
             "blob_size": len(in_blob),
         }
         log("Sending encode request")
@@ -275,7 +280,7 @@ class RemoteCLIPProxy:
             "gen_kwargs": gen_kwargs,
             "lora_stack": lora_stack,
             "tensor_inputs": meta,
-            "transport_dtype": self.transport_dtype,
+            "transport_dtype": self.transport_dtype_name,
             "blob_size": len(blob),
         }
         log("Sending generate request")