fix tests (#335)

This commit is contained in:
Christopher Hesse
2019-04-20 15:06:18 -07:00
committed by Peter Zhokhov
parent 967fc8c37f
commit bc4eef6053

View File

@@ -92,7 +92,6 @@ class JSONOutputFormat(KVWriter):
def writekvs(self, kvs): def writekvs(self, kvs):
for k, v in sorted(kvs.items()): for k, v in sorted(kvs.items()):
if hasattr(v, 'dtype'): if hasattr(v, 'dtype'):
v = v.tolist()
kvs[k] = float(v) kvs[k] = float(v)
self.file.write(json.dumps(kvs) + '\n') self.file.write(json.dumps(kvs) + '\n')
self.file.flush() self.file.flush()