Function check_key

mlx/data/Sample.cpp:18–29

}

std::shared_ptr<Array>
check_key(const Sample& input, const std::string& key, ArrayType type) {
  auto it = input.find(key);
  if (it == input.end()) {
    throw std::runtime_error("key <" + key + "> expected");
  }
  auto value = it->second;
  if (type != ArrayType::Any && value->type() != type) {
    throw std::runtime_error("invalid Array type");
  }
  return value;
}
} // namespace sample
} // namespace data
} // namespace mlx