model/pde_1d_surrogate_model/burgers_operator.py missing package "pde_gen_control".

[guracat]

`import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
from torch.utils.data.dataset import random_split

import numpy as np
import random
import os
import sys
import matplotlib.pyplot as plt
sys.path.append(os.path.join(os.path.dirname("file"), '..'))
sys.path.append(os.path.join(os.path.dirname("file"), '..', '..'))
sys.path.append(os.path.join(os.path.dirname("file"), '..', '..', '..'))
try:
from scripts.utils import *
from scripts.burgers_numeric import *
from scripts.models import *
except:
from .scripts.utils import *
from .scripts.burgers_numeric import *
from .scripts.models import *
from pde_gen_control.data_burgers_1d import Burgers1DSimple, Burgers`

When run python /model/pde_1d_surrogate_model/burgers_operator.py --date_time "2024-01-09_1d_surrogate_partial_ob_partial_ctr_autoregress-3" --epochs 500 --gpu 0 --train_batch_size 151 --autoregress_steps 3 --dataset_path "/dataset_control_burgers/free_u_f_1e5_front_rear_quarter" --is_partially_observable 1 --is_partially_controllable 1 --lr 0.001
get ModuleNotFounderror:
from pde_gen_control.data_burgers_1d import Burgers1DSimple, Burgers ModuleNotFoundError: No module named 'pde_gen_control'

[Zhangtao167]

Hello, thanks for your interest in our work. we have uploaded the missing file in https://github.com/AI4Science-WestlakeU/diffphycon/blob/main/model/pde_1d_surrogate_model/scripts/data_burgers_1d.py and updated the burgers_operator.pyhttps://github.com/AI4Science-WestlakeU/diffphycon/blob/27f33dc1a09cb255e16fb97d0b50fa919d9b213a/model/pde_1d_surrogate_model/burgers_operator.py#L24