Tachi67 commited on
Commit
7a37e36
1 Parent(s): a761f24

Update UpdatePlanAtomicFlow.py

Browse files
Files changed (1) hide show
  1. UpdatePlanAtomicFlow.py +1 -1
UpdatePlanAtomicFlow.py CHANGED
@@ -1,6 +1,6 @@
1
  #TODO: generalize updateplanatomicflow with the one in extendlibrary
2
  from typing import Dict, Any
3
- from flows.base_flows.atomic import AtomicFlow
4
  class UpdatePlanAtomicFlow(AtomicFlow):
5
  def _check_input(self, input_data: Dict[str, Any]):
6
  assert "memory_files" in input_data, "memory_files not passed to UpdatePlanAtomicFlow.yaml"
 
1
  #TODO: generalize updateplanatomicflow with the one in extendlibrary
2
  from typing import Dict, Any
3
+ from aiflows.base_flows.atomic import AtomicFlow
4
  class UpdatePlanAtomicFlow(AtomicFlow):
5
  def _check_input(self, input_data: Dict[str, Any]):
6
  assert "memory_files" in input_data, "memory_files not passed to UpdatePlanAtomicFlow.yaml"