union commit

src/data/dataset.py

@@ -1,3 +1,4 @@
+# src/data/dataset.py
 import torch
 from torch_geometric.data import Dataset, InMemoryDataset
 import os

src/data/gds_parser.py

@@ -1,3 +1,4 @@
+# src/data/gds_parser.py
 from typing import List, Dict, Tuple
 import gdstk
 import numpy as np

src/data/graph_constructor.py

@@ -1,3 +1,4 @@
+# src/data/graph_constructor.py
 from typing import List, Dict, Tuple
 import torch
 from torch_geometric.data import Data

src/data/init.py

@@ -0,0 +1 @@
+# src/data/init.py

src/engine/evaluator.py

@@ -1,3 +1,4 @@
+# src/engine/evaluator.py
 import torch
 from torch_geometric.data import DataLoader
 from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, roc_auc_score

src/engine/init.py

@@ -0,0 +1 @@
+# src/engine/init.py

src/engine/self_supervised.py

@@ -1,3 +1,4 @@
+# src/engine/self_supervised.py
 import torch
 import torch.nn as nn
 from torch.optim import AdamW

src/engine/trainer.py

@@ -1,3 +1,4 @@
+# src/engine/trainer.py
 import torch
 import torch.nn as nn
 from torch.optim import Adam, AdamW

src/init.py

@@ -0,0 +1 @@
+# src/init.py

src/models/geo_layout_transformer.py

@@ -1,3 +1,4 @@
+# src/models/geo_layout_transformer.py
 import torch
 import torch.nn as nn
 from .gnn_encoder import GNNEncoder

src/models/gnn_encoder.py

@@ -1,3 +1,4 @@
+# src/models/gnn_encoder.py
 import torch
 import torch.nn as nn
 from torch_geometric.nn import GCNConv, SAGEConv, GATConv, global_mean_pool

src/models/init.py

@@ -0,0 +1 @@
+# src/models/init.py

src/models/task_heads.py

@@ -1,3 +1,4 @@
+# src/models/task_heads.py
 import torch
 import torch.nn as nn
 

src/models/transformer_core.py

@@ -1,3 +1,4 @@
+# src/models/transformer_core.py
 import torch
 import torch.nn as nn
 import math

src/utils/config_loader.py

@@ -1,3 +1,4 @@
+# src/utils/config_loader.py
 import yaml
 from pathlib import Path
 

src/utils/init.py

@@ -0,0 +1 @@
+# src/utils/init.py

src/utils/logging.py

@@ -1,3 +1,4 @@
+# src/utils/logging.py
 import logging
 import sys