Merged var.h and varset.h from SVN head
[libdai.git] / src / mf.cpp
index 1dd0cfd..1ffc360 100644 (file)
@@ -190,7 +190,7 @@ Complex MF::logZ() const {
 
 void MF::init( const VarSet &ns ) {
     for( size_t i = 0; i < nrVars(); i++ ) {
-        if( ns && var(i) )
+        if( ns.contains(var(i) ) )
             _beliefs[i].fill( 1.0 );
     }
 }