From cd19c5487604504b57c8fd8211dbaa5105d611d9 Mon Sep 17 00:00:00 2001 From: ayxan Date: Wed, 23 Nov 2022 00:39:06 +0400 Subject: [PATCH] added typing.Union --- src/data_access/data_opeartions.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/data_access/data_opeartions.py b/src/data_access/data_opeartions.py index e908508..45168b6 100644 --- a/src/data_access/data_opeartions.py +++ b/src/data_access/data_opeartions.py @@ -2,21 +2,22 @@ import pandas as pd from datetime import (datetime, timedelta) from typing import (List, - Tuple) + Tuple, + Union) class DataOperations: heart_rate = None def is_data_none_wrapper(func): - def __is_data_none(self, *args, **kwargs) -> None | NameError: + def __is_data_none(self, *args, **kwargs) -> Union[None, NameError]: if self.data is None: raise NameError(f'data not found. You must call the get_data function before calling the {func.__name__} function') return func(self, *args, **kwargs) return __is_data_none def is_heart_rate_none_wrapper(func): - def __is_heart_rate_none(self, *args, **kwargs) -> None | NameError: + def __is_heart_rate_none(self, *args, **kwargs) -> Union[None, NameError]: if self.heart_rate is None: raise NameError(f'heart_rate data not found. You must call the get_heart_rate function before calling the {func.__name__} function') return func(self, *args, **kwargs)