ENH: add skip_mode to all restoreModel functions

ExaScience · Feb 6, 2019 · 5e7ba4e · 5e7ba4e
1 parent 68f3cab
commit 5e7ba4e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp b/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp
@@ -238,9 +238,9 @@ std::shared_ptr<Model> PredictSession::restoreModel(const std::shared_ptr<StepFi
     return model;
 }
 
-std::shared_ptr<Model> PredictSession::restoreModel(int i)
+std::shared_ptr<Model> PredictSession::restoreModel(int i, int skip_mode)
 {
-    return restoreModel(m_stepfiles.at(i));
+    return restoreModel(m_stepfiles.at(i), skip_mode);
 }
 
 // predict one element

diff --git a/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h b/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h
@@ -40,7 +40,7 @@ class PredictSession : public ISession
 
 private:
     std::shared_ptr<Model> restoreModel(const std::shared_ptr<StepFile> &, int skip_mode = -1);
-    std::shared_ptr<Model> restoreModel(int i);
+    std::shared_ptr<Model> restoreModel(int i, int skip_mode = -1);
 
 public:
     int    getNumSteps()  const { return m_stepfiles.size(); }