code: def query_all_model_ids(self, dataset):
      if self.space_name == Config.NB101:
          self.api = Gt101()
      elif self.space_name == Config.NB201:
          self.api = Gt201()
      elif self.space_name == Config.MLPSP:
          self.api = GTMLP(dataset)
      return self.api.get_all_trained_model_ids()
  
file_path: singa/examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/src/query_api/interface.py
imports: from src.common.constant import Config, CommonVars
  from src.query_api.query_api_img import Gt201, Gt101
  from src.query_api.query_api_mlp import GTMLP
  from src.query_api.query_api_img import ImgScoreQueryApi
  from typing import *
label_desc: Use image node_id 5 for calling the SimulateTrain obj's underlying member method code with example usage: obj.query_all_model_ids(dataset) and returns: self