Formatting adjustments.

hydrogym/core.py

@@ -195,13 +195,15 @@ def render(self, **kwargs):
         """Plot the current PDE state (called by `gym.Env`)"""
         raise NotImplementedError
 
+
 '''
 
 @ray.remote
 class EvaluationActor:
     """To remotely evaluate Firedrake solutions."""
 
-    def __init__(self, firedrake_instance: "Firedrake_instance", index: int, seeds: Union[ActorSeeds, tuple], state:dict):
+    def __init__(self, firedrake_instance: "Firedrake_instance", index: int, seeds:
+            Union[ActorSeeds, tuple], state:dict):
         """
         Initialize a remote runner for a Firedrake problem instance.
 
@@ -217,6 +219,7 @@ def __init__(self, firedrake_instance: "Firedrake_instance", index: int, seeds:
        #   code here running.
 '''
 
+
 class CallbackBase:
     def __init__(self, interval: int = 1):
         """

hydrogym/firedrake/flow.py

@@ -237,7 +237,9 @@ def control_vec(self, mixed=False):
 
             # Control as Function
             B.append(
-                fd.assemble(inner(fd.Constant((0, 0)), v) * dx, bcs=self.collect_bcs()).riesz_representation(riesz_map = 'l2')
+                fd.assemble(
+                    inner(fd.Constant((0, 0)), v) * dx, cs=self.collect_bcs()
+                ).riesz_representation(riesz_map="l2")
             )
 
             # Have to have mixed function space for computing B functions