Skip to content

Commit 5e7ba4e

Browse files
committed
ENH: add skip_mode to all restoreModel functions
1 parent 68f3cab commit 5e7ba4e

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,9 +238,9 @@ std::shared_ptr<Model> PredictSession::restoreModel(const std::shared_ptr<StepFi
238238
return model;
239239
}
240240

241-
std::shared_ptr<Model> PredictSession::restoreModel(int i)
241+
std::shared_ptr<Model> PredictSession::restoreModel(int i, int skip_mode)
242242
{
243-
return restoreModel(m_stepfiles.at(i));
243+
return restoreModel(m_stepfiles.at(i), skip_mode);
244244
}
245245

246246
// predict one element

lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class PredictSession : public ISession
4040

4141
private:
4242
std::shared_ptr<Model> restoreModel(const std::shared_ptr<StepFile> &, int skip_mode = -1);
43-
std::shared_ptr<Model> restoreModel(int i);
43+
std::shared_ptr<Model> restoreModel(int i, int skip_mode = -1);
4444

4545
public:
4646
int getNumSteps() const { return m_stepfiles.size(); }

0 commit comments

Comments
 (0)