Add input and output dim

MeasureTransport · Feb 6, 2024 · b114ca9 · b114ca9
1 parent 8195c60
commit b114ca9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/MParT.jl b/src/MParT.jl
@@ -7,7 +7,7 @@ module MParT
     import Base: getindex, lastindex, show, iterate, convert
 
     ConditionalMapBasePtr = CxxWrap.StdLib.SharedPtr{<:ConditionalMapBase}
-    for op = (:Evaluate, :Gradient, :Inverse,
+    for op = (:Evaluate, :Gradient, :Inverse, :inputDim, :outputDim,
               :LogDeterminant, :LogDeterminantCoeffGrad, :LogDeterminantInputGrad,
               :numCoeffs, :CoeffMap, :CoeffGrad, :SetCoeffs, :TestError)
         eval(quote