Skip to content

Commit

Permalink
formatting cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
attila-balint-kul committed Jun 22, 2023
1 parent e76a088 commit c3ed3da
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 6 deletions.
5 changes: 2 additions & 3 deletions src/enfobench/evaluation/_cross_validate.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
from typing import Union, List, Optional
from typing import List, Optional, Union

import pandas as pd
from tqdm import tqdm

from enfobench.utils import steps_in_horizon

from enfobench.evaluation.client import ForecastClient
from enfobench.evaluation.protocols import Model
from enfobench.utils import steps_in_horizon


def generate_cutoff_dates(
Expand Down
4 changes: 2 additions & 2 deletions src/enfobench/evaluation/client.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import io
from typing import Dict, Union, Optional, List
from typing import Dict, List, Optional, Union

import pandas as pd
import requests
Expand Down Expand Up @@ -46,7 +46,7 @@ def predict(
if level is not None:
params["level"] = level

y_df = y.rename_axis('ds').reset_index()
y_df = y.rename_axis("ds").reset_index()
files = {
"y": to_buffer(y_df),
# "X": to_buffer(X),
Expand Down
2 changes: 1 addition & 1 deletion src/enfobench/evaluation/protocols.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from enum import Enum
from typing import Any, Protocol, Optional, List, Dict
from typing import Any, Dict, List, Optional, Protocol

import pandas as pd
from pydantic import BaseModel
Expand Down

0 comments on commit c3ed3da

Please sign in to comment.