Update import.

aiidaplugins · Dec 20, 2023 · 8e09392 · 8e09392
1 parent c5371c9
commit 8e09392
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/examples/example_workflow_combi.py b/examples/example_workflow_combi.py
@@ -6,8 +6,7 @@
 import datetime
 import click
 import yaml
-from aiida import orm, plugins, engine, cmdline
-from aiida.common.datastructures import StashMode
+from aiida import orm, plugins, engine, cmdline, common
 from aiida_flexpart.utils import reformat_locations
 
 
@@ -177,21 +176,21 @@ def test_run(flexpart_code):
         'source_list':
         ['aiida.out', 'header*', 'partposit_inst', 'grid_time_*.nc'],
         'target_base': f'/store/empa/em05/{username}/aiida_stash',
-        'stash_mode': StashMode.COPY.value,
+        'stash_mode': common.StashMode.COPY.value,
     }
     builder.flexpartifs.metadata.options.stash = {
         'source_list':
         ['aiida.out', 'header*', 'partposit_inst*', 'grid_time_*.nc'],
         'target_base':
         f'/store/empa/em05/{username}/aiida_stash',
         'stash_mode':
-        StashMode.COPY.value,
+        common.StashMode.COPY.value,
     }
     builder.flexpartpost.metadata.options.stash = {
         'source_list':
         ['aiida.out', 'boundary_sensitivity_*.nc', 'grid_time_*.nc'],
         'target_base': f'/store/empa/em05/{username}/aiida_stash',
-        'stash_mode': StashMode.COPY.value,
+        'stash_mode': common.StashMode.COPY.value,
     }
 
     #change wall time for cosmo and ifs in seconds