Simple fix for mkl_multiclass crasher

shogun-toolbox · May 22, 2012 · 6216379 · 6216379
1 parent 52d53fd
commit 6216379
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/shogun/machine/MulticlassMachine.cpp b/src/shogun/machine/MulticlassMachine.cpp
@@ -52,7 +52,8 @@ CMulticlassMachine::~CMulticlassMachine()
 
 void CMulticlassMachine::set_labels(CLabels* lab)
 {
-	ASSERT(lab->get_label_type() == LT_MULTICLASS);
+    if (lab)
+        ASSERT(lab->get_label_type() == LT_MULTICLASS);
     CMachine::set_labels(lab);
     if (lab)
         init_strategy();