File tree Expand file tree Collapse file tree 4 files changed +3
-221
lines changed
Expand file tree Collapse file tree 4 files changed +3
-221
lines changed Original file line number Diff line number Diff line change 2626import functools
2727import logging
2828import os
29- from typing import Callable
30- from typing import Final
29+ from typing import Callable , Final
3130
3231import chex
3332import jax
3433from jax import numpy as jnp
35-
3634from torax import geometry
3735from torax import state
3836from torax .config import runtime_params_slice
4240from torax .transport_model import runtime_params as runtime_params_lib
4341from torax .transport_model import transport_model
4442
43+
4544# Environment variable for the QLKNN model. Used if the model path
4645# is not set in the config.
4746MODEL_PATH_ENV_VAR : Final [str ] = 'TORAX_QLKNN_MODEL_PATH'
Original file line number Diff line number Diff line change 2121
2222from __future__ import annotations
2323
24+ from collections .abc import Callable
2425import dataclasses
2526import datetime
2627import os
2728import subprocess
2829import tempfile
29- from collections .abc import Callable
3030
3131import chex
3232import numpy as np
3333from qualikiz_tools .qualikiz_io import inputfiles as qualikiz_inputtools
3434from qualikiz_tools .qualikiz_io import qualikizrun as qualikiz_runtools
35-
3635from torax import geometry
3736from torax import jax_utils
3837from torax import state
Load Diff This file was deleted.
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments