merge

src/library/variational_models.py

@@ -215,7 +215,7 @@ class GaussianEntropy(EnergyOperator):
 
     Parameters
     ----------
-    domain: Domain FIXME
+    domain: Domain, DomainTuple, list of Domain
         The domain of the diagonal.
     """
 
@@ -244,7 +244,7 @@ class LowerTriangularInserter(LinearOperator):
 
     Parameters
     ----------
-    target: Domain FIXME
+    target: Domain, DomainTuple, list of Domain
         A two-dimensional domain with NxN entries.
     """
 
@@ -273,7 +273,7 @@ class DiagonalSelector(LinearOperator):
 
     Parameters
     ----------
-    domain: Domain FIXME
+    domain: Domain, DomainTuple, list of Domain
         The two-dimensional domain of the input field. Must be of shape NxN.
     """
 

src/operators/multifield2vector.py

@@ -27,9 +27,14 @@ class Multifield2Vector(LinearOperator):
     """Flatten a MultiField and return a Field with unstructured domain and the
     same number of degrees of freedom.
 
+    Parameters
+    ----------
+    domain: MultiDomain
+        Domain of the operator
+
     Notes
     -----
-    Currently only works with fields that are real-valued
+    Currently only works with fields that are real-valued.
     """
 
     def __init__(self, domain):