Skip to content

Commit

Permalink
Include new args in comment
Browse files Browse the repository at this point in the history
  • Loading branch information
corepattern authored Oct 3, 2023
1 parent 05f4f2e commit 5242965
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions torchquad/integration/grid_integrator.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,6 +151,7 @@ def get_jit_compiled_integrate(
N (int, optional): Total number of sample points to use for the integration. See the integrate method documentation for more details.
integration_domain (list or backend tensor, optional): Integration domain, e.g. [[-1,1],[0,1]]. Defaults to [-1,1]^dim. It can also determine the numerical backend.
backend (string, optional): Numerical backend. Defaults to integration_domain's backend if it is a tensor and otherwise to the backend from the latest call to set_up_backend or "torch" for backwards compatibility.
args (list or tuple, optional): Any arguments required by the function. Defaults to None.
Returns:
function(fn, integration_domain): JIT compiled integrate function where all parameters except the integrand and domain are fixed
Expand Down

0 comments on commit 5242965

Please sign in to comment.