Update SQLInferenceDataStore.scala