def load_countgraph(filename, small=False): """Load a countgraph object from the given filename and return it. Keyword argument: filename -- the name of the countgraph file small -- set this to load a SmallCountgraph instance """ if small: countgraph = _SmallCountgraph(1, [1]) countgraph.load(filename) else: countgraph = _Countgraph(1, [1]) countgraph.load(filename) return countgraph