CollapsedDocStrings = true Argmax2D Public Modules = [DecisionFocusedLearningBenchmarks.Argmax2D] Private = false Private Modules = [DecisionFocusedLearningBenchmarks.Argmax2D] Public = false