From ac1f0817c61011db434d9c188517b4cb766a6a0f Mon Sep 17 00:00:00 2001
From: ellis <eshoag2@illinois.edu>
Date: Wed, 15 Feb 2017 14:29:24 -0600
Subject: [PATCH] pymetis dependency
---
meshmode/mesh/processing.py | 10 ++++++----
requirements.txt | 3 +++
2 files changed, 9 insertions(+), 4 deletions(-)
diff --git a/meshmode/mesh/processing.py b/meshmode/mesh/processing.py
index 28d146b7..8a6f0464 100644
--- a/meshmode/mesh/processing.py
+++ b/meshmode/mesh/processing.py
@@ -90,11 +90,13 @@ def partition_mesh(mesh, part_per_element, part_nr):
new_indices.append(mesh_group.vertex_indices[elems])
new_nodes.append(
- np.zeros((mesh.ambient_dim, end_idx - start_idx, mesh_group.nunit_nodes)))
+ np.zeros(
+ (mesh.ambient_dim, end_idx - start_idx, mesh_group.nunit_nodes)))
for i in range(mesh.ambient_dim):
for j in range(start_idx, end_idx):
elems = queried_elems[j] - num_prev_elems
- new_nodes[group_nr][i, j - start_idx, :] = mesh_group.nodes[i, elems, :]
+ new_idx = j - start_idx
+ new_nodes[group_nr][i, new_idx, :] = mesh_group.nodes[i, elems, :]
index_set |= set(new_indices[group_nr].ravel())
@@ -110,7 +112,7 @@ def partition_mesh(mesh, part_per_element, part_nr):
# Our indices need to be in range [0, len(mesh.nelements)].
for group_nr in range(num_groups):
- if not group_nr in skip_groups:
+ if group_nr not in skip_groups:
for i in range(len(new_indices[group_nr])):
for j in range(len(new_indices[group_nr][0])):
original_index = new_indices[group_nr][i, j]
@@ -121,7 +123,7 @@ def partition_mesh(mesh, part_per_element, part_nr):
new_mesh_groups = []
for group_nr in range(num_groups):
- if not group_nr in skip_groups:
+ if group_nr not in skip_groups:
mesh_group = mesh.groups[group_nr]
new_mesh_groups.append(
SimplexElementGroup(mesh_group.order, new_indices[group_nr],
diff --git a/requirements.txt b/requirements.txt
index 6ed102e6..f5bb3464 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,3 +9,6 @@ git+https://github.com/inducer/loopy.git
git+https://github.com/inducer/boxtree.git
git+https://github.com/inducer/sumpy.git
git+https://github.com/inducer/pytential.git
+
+# requires pymetis for tests for partition_mesh
+git+https://github.com/inducer/pymetis.git
--
GitLab