#!/usr/bin/env python3
"""
Experiment script for DLinear model
"""

import sys
import runpy

# DLinear experiment configuration
sys.argv = [
    "main.py",
    "--dataset", "electricity",
    "--num_clients", "20",
    "--max_features", "20",
    "--model_name", "dlinear",
    "--window", "96",
    "--horizon", "48",
    "--batch_size", "128",
    "--global_epochs", "50",
    "--local_epochs", "1",
    "--lr", "1e-3",
    "--optimizer", "adam",
    "--seed", "42",
    "--device", "cuda",
    "--experiment", "electricity_dlinear_50global_96-48_dim128",
    "--frac", "1.0",
    "--dim", "128",
    "--early_stop",
    "--patience", "10"
]

if __name__ == "__main__":
    # Change to parent directory to run main.py
    import os
    os.chdir("..")
    runpy.run_path("main.py", run_name="__main__")