docstr: Creates ProblemFactory for NASBench201.
  
  Args:
      dataset:
          Accepts one of "cifar10", "cifar100" or "ImageNet16-120".
docstr_tok: ["Creates","ProblemFactory","for","NASBench201",".","Args",":","dataset",":","Accepts","one","of","``","cifar10","''",",","``","cifar100","''","or","``","ImageNet16-120","''","."]
code: def __init__(self, dataset: str) -> None:
      """Creates ProblemFactory for NASBench201.
  
      Args:
          dataset:
              Accepts one of "cifar10", "cifar100" or "ImageNet16-120".
      """
      self._dataset = dataset
      if dataset == "cifar10":
          # Set name used in dataset API.
          self._dataset = "cifar10-valid"
      self._dataset_api = get_dataset_api("nasbench201", dataset)
  
file_path: optuna/benchmarks/naslib/problem.py
imports: from __future__ import annotations
  import sys
  from typing import Any
  from kurobako import problem
  from naslib.utils import get_dataset_api
label_desc: Use image node_id 1 to create a new NASLibProblemFactory object from inherited base classes: problem with example: obj = NASLibProblemFactory(dataset)