This commit is contained in:
Richard Liaw
2017-10-01 22:34:55 -07:00
committed by Philipp Moritz
parent 2892955119
commit 54aaeb25ab
+3 -1
View File
@@ -22,11 +22,13 @@ logger.setLevel(logging.INFO)
class RLLibEncoder(json.JSONEncoder):
def default(self, value):
if isinstance(value, np.float32) or isinstance(value, np.float64):
if np.issubdtype(value, float):
if np.isnan(value):
return None
else:
return float(value)
elif np.issubdtype(value, int):
return int(value)
class RLLibLogger(object):