Commit 5090c125 authored by Pierre Paleo's avatar Pierre Paleo

Fix frame_size in options

parent dcd04d01
......@@ -24,7 +24,11 @@ class MemcachedClient(object):
config = default_config(Client_config)
config.update(client_params)
self.params = config
self.frame_shape = self.params["request_size"]
rs = self.params["request_size"]
if isinstance(rs, str):
self.frame_shape = tuple(map(int, rs.replace("(", "").replace(")", "").split(",")))
else:
self.frame_shape = rs
self.frame_dtype = np.uint16 # TODO custom ?
self.frame_bytes = np.iinfo(self.frame_dtype).bits //8 * np.prod(self.frame_shape)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment