Commit d65cfbf7 authored by jay's avatar jay
Browse files

Adds the np encoder for JSON arrays

parent 71909a32
Loading
Loading
Loading
Loading
+9 −0
Original line number Diff line number Diff line
import json
import numpy as np


class NumpyEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.ndarray):
            return obj.tolist()
        elif isinstance(obj, datetime.date):
            return obj.isoformat()
        return json.JSONEncoder.default(self, obj)

def read_json(inputfile):
    """
    Read the input json file into a python dictionary.