From 3fe3fa407bdde687eedc68d2e6f94ad7b8271fbf Mon Sep 17 00:00:00 2001
From: Thomas Gibson <gibsonthomas1120@hotmail.com>
Date: Thu, 29 Apr 2021 22:42:35 -0500
Subject: [PATCH] Clean up flake8 errors in geometry module
---
grudge/geometry/__init__.py | 7 +++++++
grudge/geometry/metrics.py | 6 ++++--
2 files changed, 11 insertions(+), 2 deletions(-)
diff --git a/grudge/geometry/__init__.py b/grudge/geometry/__init__.py
index c5a83ba5..02dbc5f1 100644
--- a/grudge/geometry/__init__.py
+++ b/grudge/geometry/__init__.py
@@ -29,3 +29,10 @@ from grudge.geometry.metrics import (
inverse_surface_metric_derivative,
area_element
)
+
+__all__ = (
+ "forward_metric_derivative_mat",
+ "inverse_metric_derivative_mat",
+ "inverse_surface_metric_derivative",
+ "area_element"
+)
diff --git a/grudge/geometry/metrics.py b/grudge/geometry/metrics.py
index c4f66da2..44f741cd 100644
--- a/grudge/geometry/metrics.py
+++ b/grudge/geometry/metrics.py
@@ -140,7 +140,8 @@ def inverse_metric_derivative_mat(actx, dcoll, dd=None):
@memoize_on_first_arg
def inverse_first_fundamental_form(actx, dcoll, dd):
- if dcoll.ambient_dim == dcoll.dim:
+ dim = dcoll.dim
+ if dcoll.ambient_dim == dim:
inv_mder = inverse_metric_derivative_mat(actx, dcoll, dd)
inv_form1 = inv_mder.dot(inv_mder.T)
else:
@@ -201,7 +202,8 @@ def inverse_metric_derivative(actx, dcoll, rst_axis, xyz_axis, dd):
@memoize_on_first_arg
def inverse_surface_metric_derivative(actx, dcoll, rst_axis, xyz_axis, dd=None):
- if dcoll.ambient_dim == dcoll.dim:
+ dim = dcoll.dim
+ if dcoll.ambient_dim == dim:
imd = inverse_metric_derivative(
actx, dcoll, rst_axis, xyz_axis, dd=dd
)
--
GitLab