1 files changed, 4 insertions, 3 deletions
diff --git a/tests/InferenceTest.hpp b/tests/InferenceTest.hpp
index 91a65ea494..3ebfdbcc3c 100644
--- a/tests/InferenceTest.hpp
+++ b/tests/InferenceTest.hpp
@@ -91,7 +91,7 @@ public:
     virtual ~IInferenceTestCaseProvider() {}
 
     virtual void AddCommandLineOptions(boost::program_options::options_description& options) {};
-    virtual bool ProcessCommandLineOptions() { return true; };
+    virtual bool ProcessCommandLineOptions(const InferenceTestOptions &commonOptions) { return true; };
     virtual std::unique_ptr<IInferenceTestCase> GetTestCase(unsigned int testCaseId) = 0;
     virtual bool OnInferenceTestFinished() { return true; };
 };
@@ -219,7 +219,7 @@ public:
     ClassifierTestCaseProvider(TConstructDatabaseCallable constructDatabase, TConstructModelCallable constructModel);
 
     virtual void AddCommandLineOptions(boost::program_options::options_description& options) override;
-    virtual bool ProcessCommandLineOptions() override;
+    virtual bool ProcessCommandLineOptions(const InferenceTestOptions &commonOptions) override;
     virtual std::unique_ptr<IInferenceTestCase> GetTestCase(unsigned int testCaseId) override;
     virtual bool OnInferenceTestFinished() override;
 
@@ -227,7 +227,8 @@ private:
     void ReadPredictions();
 
     typename InferenceModel::CommandLineOptions m_ModelCommandLineOptions;
-    std::function<std::unique_ptr<InferenceModel>(typename InferenceModel::CommandLineOptions)> m_ConstructModel;
+    std::function<std::unique_ptr<InferenceModel>(const InferenceTestOptions& commonOptions,
+                                                  typename InferenceModel::CommandLineOptions)> m_ConstructModel;
     std::unique_ptr<InferenceModel> m_Model;
 
     std::string m_DataDir;