Standardize E2E module names. (#2872)
The class names of the E2E modules are used to index into their benchmarking artifacts, so they need to be distinct. This is currently untested, and there was a collision between `depth_conv_test` and `conv_test` since they both used `Conv2dModule`. This fixes that and gives other modules more specific names as a stopgap.
Additional NFC: use `os.makedirs(exist_ok=True)` instead of `try`/`catch`.
diff --git a/integrations/tensorflow/e2e/dynamic_mlp_test.py b/integrations/tensorflow/e2e/dynamic_mlp_test.py
index 0b70e84..ff905a5 100644
--- a/integrations/tensorflow/e2e/dynamic_mlp_test.py
+++ b/integrations/tensorflow/e2e/dynamic_mlp_test.py
@@ -24,7 +24,7 @@
CLASSES = 10
-class Mlp(tf.Module):
+class DynamicMlpModule(tf.Module):
def __init__(self,
hidden_1_dim=256,
@@ -61,7 +61,7 @@
return tf.nn.softmax(self.mlp(x))
-@tf_test_utils.compile_module(Mlp, exported_names=["predict"])
+@tf_test_utils.compile_module(DynamicMlpModule, exported_names=["predict"])
class DynamicMlpTest(tf_test_utils.TracedModuleTestCase):
def test_dynamic_batch(self):
