Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jun 28, 2022
1 parent e9dd036 commit 85df4dd
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions ludwig/data/postprocessing.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,15 +14,15 @@
# limitations under the License.
# ==============================================================================
import os
from typing import Any, Dict, List, Union, Optional
from typing import Any, Dict, List, Optional, Union

import numpy as np
import pandas as pd
import torch

from ludwig.backend import LOCAL_BACKEND
from ludwig.utils.data_utils import DATAFRAME_FORMATS, DICT_FORMATS
from ludwig.utils.dataframe_utils import to_numpy_dataset, is_dask_backend
from ludwig.utils.dataframe_utils import is_dask_backend, to_numpy_dataset
from ludwig.utils.misc_utils import get_from_registry
from ludwig.utils.strings_utils import make_safe_filename

Expand Down Expand Up @@ -105,11 +105,7 @@ def convert_dict_to_df(predictions: Dict[str, Dict[str, Union[List[Any], torch.T

def convert_predictions(predictions, output_features, return_type="dict", backend: Optional["Backend"] = None):
convert_fn = get_from_registry(return_type, conversion_registry)
return convert_fn(
predictions,
output_features,
backend
)
return convert_fn(predictions, output_features, backend)


def convert_to_dict(
Expand Down

0 comments on commit 85df4dd

Please sign in to comment.