def METHOD_NAME(self):
    graph = NasBench201SearchSpace()
    graph.sample_random_architecture(dataset_api=create_dummy_api())
    try:
        results = graph.query(metric=Metric.VAL_ACCURACY, dataset='cifar10')
    except Exception as e:
        assert isinstance(e, NotImplementedError)