class ProblemInference: def infer(self, dataset_info, target_column): if dataset_info.get("possible_nlp"): return "nlp" if target_column: return "classification" if dataset_info.get("class_distribution") else "regression" return "unknown"