Fix input argument error with generic plugin

nasa · Sep 11, 2023 · 266c512 · 266c512
1 parent d91a21e
commit 266c512
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugins/generic/generic_plugin.py b/plugins/generic/generic_plugin.py
@@ -11,13 +11,13 @@
 from onair.src.data_driven_components.ai_plugin_abstract.core import AIPlugIn
 
 class Plugin(AIPlugIn):
- def apriori_training(self):
+ def apriori_training(self,batch_data=[]):
  """
  Given data, system should learn any priors necessary for realtime diagnosis.
  """
  pass
 
- def update(self):
+ def update(self,frame=[]):
  """
  Given streamed data point, system should update internally
  """