99b31de8 · 99b31de8 · 99b31de8 · 99b31de8 · 99b31de8 · 99b31de8
--- a/tracegnn/models/trace_vae/constants.py
+++ b/tracegnn/models/trace_vae/constants.py
+from tracegnn.constants import *
+
+LATENCY_DIM = 1
--- a/tracegnn/models/trace_vae/dataset.py
+++ b/tracegnn/models/trace_vae/dataset.py
--- a/tracegnn/models/trace_vae/distributions.py
+++ b/tracegnn/models/trace_vae/distributions.py
--- a/tracegnn/models/trace_vae/evaluation.py
+++ b/tracegnn/models/trace_vae/evaluation.py
--- a/tracegnn/models/trace_vae/graph_utils.py
+++ b/tracegnn/models/trace_vae/graph_utils.py
--- a/tracegnn/models/trace_vae/model/__init__.py
+++ b/tracegnn/models/trace_vae/model/__init__.py
--- a/tracegnn/models/trace_vae/model/gnn_layers.py
+++ b/tracegnn/models/trace_vae/model/gnn_layers.py
--- a/tracegnn/models/trace_vae/model/latency_vae.py
+++ b/tracegnn/models/trace_vae/model/latency_vae.py
--- a/tracegnn/models/trace_vae/model/model_utils.py
+++ b/tracegnn/models/trace_vae/model/model_utils.py
--- a/tracegnn/models/trace_vae/model/operation_embedding.py
+++ b/tracegnn/models/trace_vae/model/operation_embedding.py
--- a/tracegnn/models/trace_vae/model/pooling.py
+++ b/tracegnn/models/trace_vae/model/pooling.py
--- a/tracegnn/models/trace_vae/model/realnvp_flow.py
+++ b/tracegnn/models/trace_vae/model/realnvp_flow.py
--- a/tracegnn/models/trace_vae/model/struct_vae.py
+++ b/tracegnn/models/trace_vae/model/struct_vae.py
--- a/tracegnn/models/trace_vae/model/trace_vae.py
+++ b/tracegnn/models/trace_vae/model/trace_vae.py
--- a/tracegnn/models/trace_vae/tensor_utils.py
+++ b/tracegnn/models/trace_vae/tensor_utils.py
--- a/tracegnn/models/trace_vae/test.py
+++ b/tracegnn/models/trace_vae/test.py
--- a/tracegnn/models/trace_vae/test_utils.py
+++ b/tracegnn/models/trace_vae/test_utils.py
--- a/tracegnn/models/trace_vae/train.py
+++ b/tracegnn/models/trace_vae/train.py
--- a/tracegnn/models/trace_vae/types.py
+++ b/tracegnn/models/trace_vae/types.py
--- a/tracegnn/utils/__init__.py
+++ b/tracegnn/utils/__init__.py