Skip to content
Snippets Groups Projects
Commit 5090c125 authored by Pierre Paleo's avatar Pierre Paleo
Browse files

Fix frame_size in options

parent dcd04d01
Branches master
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment