diff --git a/flatisfy/config.py b/flatisfy/config.py index 2211944..bd5fe52 100644 --- a/flatisfy/config.py +++ b/flatisfy/config.py @@ -143,7 +143,7 @@ def validate_config(config, check_with_data): assert config["smtp_to"] is None or isinstance(config["smtp_to"], list) assert isinstance(config["store_personal_data"], bool) - assert isinstance(config["max_distance_housing_station"], int) or isinstance(config["max_distance_housing_station"], float) # noqa: E501 + assert isinstance(config["max_distance_housing_station"], (int, float)) # Ensure constraints are ok assert config["constraints"] diff --git a/flatisfy/models/flat.py b/flatisfy/models/flat.py index c3a68a9..340e7a6 100644 --- a/flatisfy/models/flat.py +++ b/flatisfy/models/flat.py @@ -5,10 +5,10 @@ This modules defines an SQLAlchemy ORM model for a flat. # pylint: disable=locally-disabled,invalid-name,too-few-public-methods from __future__ import absolute_import, print_function, unicode_literals +import enum import logging import arrow -import enum from sqlalchemy import ( Column, DateTime, Enum, Float, SmallInteger, String, Text, inspect