Import VTKFile at top level

thetisproject · Jun 7, 2024 · 576f2a0 · 576f2a0
1 parent 1fdc2b5
commit 576f2a0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/channel_inversion/inverse_problem.py b/examples/channel_inversion/inverse_problem.py
@@ -1,6 +1,6 @@
 from thetis import *
+import firedrake as fd
 from firedrake.adjoint import *
-from firedrake.output.vtk_output import VTKFile
 import numpy
 import thetis.inversion_tools as inversion_tools
 from model_config import construct_solver
@@ -134,4 +134,4 @@
     oc.rename(name)
     print_function_value_range(oc, prefix='Optimal')
     if not no_exports:
-        VTKFile(f'{options.output_directory}/{name}_optimised.pvd').write(oc)
+        fd.VTKFile(f'{options.output_directory}/{name}_optimised.pvd').write(oc)
diff --git a/thetis/inversion_tools.py b/thetis/inversion_tools.py
@@ -80,9 +80,9 @@ def initialize(self):
             create_directory(self.output_dir + '/hdf5')
             for i in range(len(self.control_coeff_list)):
                 self.outfiles_m.append(
-                    fd.output.vtk_output.VTKFile(f'{self.output_dir}/control_progress_{i:02d}.pvd'))
+                    fd.VTKFile(f'{self.output_dir}/control_progress_{i:02d}.pvd'))
                 self.outfiles_dJdm.append(
-                    fd.output.vtk_output.VTKFile(f'{self.output_dir}/gradient_progress_{i:02d}.pvd'))
+                    fd.VTKFile(f'{self.output_dir}/gradient_progress_{i:02d}.pvd'))
         self.initialized = True
 
     def add_control(self, f):