avoid pickling environ
This commit is contained in:
parent
9b883de1cb
commit
e200fe9186
|
@ -1,7 +1,7 @@
|
||||||
from logging import getLogger
|
from logging import getLogger
|
||||||
from os import environ, path
|
from os import environ, path
|
||||||
from secrets import token_urlsafe
|
from secrets import token_urlsafe
|
||||||
from typing import Dict, List, Optional
|
from typing import List, Optional
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
|
@ -42,7 +42,6 @@ class ServerContext:
|
||||||
feature_flags: List[str]
|
feature_flags: List[str]
|
||||||
plugins: List[str]
|
plugins: List[str]
|
||||||
debug: bool
|
debug: bool
|
||||||
env: Dict[str, str]
|
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
@ -68,7 +67,6 @@ class ServerContext:
|
||||||
feature_flags: Optional[List[str]] = None,
|
feature_flags: Optional[List[str]] = None,
|
||||||
plugins: Optional[List[str]] = None,
|
plugins: Optional[List[str]] = None,
|
||||||
debug: bool = False,
|
debug: bool = False,
|
||||||
env: Dict[str, str] = environ,
|
|
||||||
) -> None:
|
) -> None:
|
||||||
self.bundle_path = bundle_path
|
self.bundle_path = bundle_path
|
||||||
self.model_path = model_path
|
self.model_path = model_path
|
||||||
|
@ -92,7 +90,6 @@ class ServerContext:
|
||||||
self.feature_flags = feature_flags or []
|
self.feature_flags = feature_flags or []
|
||||||
self.plugins = plugins or []
|
self.plugins = plugins or []
|
||||||
self.debug = debug
|
self.debug = debug
|
||||||
self.env = env
|
|
||||||
|
|
||||||
self.cache = ModelCache(self.cache_limit)
|
self.cache = ModelCache(self.cache_limit)
|
||||||
|
|
||||||
|
@ -133,7 +130,7 @@ class ServerContext:
|
||||||
)
|
)
|
||||||
|
|
||||||
def get_setting(self, flag: str, default: str) -> Optional[str]:
|
def get_setting(self, flag: str, default: str) -> Optional[str]:
|
||||||
return self.env.get(f"ONNX_WEB_{flag}", default)
|
return environ.get(f"ONNX_WEB_{flag}", default)
|
||||||
|
|
||||||
def has_feature(self, flag: str) -> bool:
|
def has_feature(self, flag: str) -> bool:
|
||||||
return flag in self.feature_flags
|
return flag in self.feature_flags
|
||||||
|
|
Loading…
Reference in New Issue