fix mypy errors after version update
This commit is contained in:
parent
8cd74a9fc4
commit
fabcbab751
3 changed files with 15 additions and 9 deletions
11
demo.py
11
demo.py
|
@ -1,6 +1,6 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
from subprocess import check_call, DEVNULL
|
from subprocess import check_call, DEVNULL
|
||||||
from shutil import copy, copytree
|
from shutil import copytree, ignore_patterns
|
||||||
import os
|
import os
|
||||||
from os.path import abspath
|
from os.path import abspath
|
||||||
from sys import executable as python
|
from sys import executable as python
|
||||||
|
@ -9,12 +9,17 @@ from pathlib import Path
|
||||||
my_repo = Path(__file__).absolute().parent
|
my_repo = Path(__file__).absolute().parent
|
||||||
|
|
||||||
|
|
||||||
def run():
|
def run() -> None:
|
||||||
# uses fixed paths; worth it for the sake of demonstration
|
# uses fixed paths; worth it for the sake of demonstration
|
||||||
# assumes we're in /tmp/my_demo now
|
# assumes we're in /tmp/my_demo now
|
||||||
|
|
||||||
# 1. clone git@github.com:karlicoss/my.git
|
# 1. clone git@github.com:karlicoss/my.git
|
||||||
copytree(my_repo, 'my_repo', symlinks=True)
|
copytree(
|
||||||
|
my_repo,
|
||||||
|
'my_repo',
|
||||||
|
symlinks=True,
|
||||||
|
ignore=ignore_patterns('.tox*'), # tox dir might have broken symlinks while tests are running in parallel
|
||||||
|
)
|
||||||
|
|
||||||
# 2. prepare repositories you'd be using. For this demo we only set up Hypothesis
|
# 2. prepare repositories you'd be using. For this demo we only set up Hypothesis
|
||||||
tox = 'TOX' in os.environ
|
tox = 'TOX' in os.environ
|
||||||
|
|
|
@ -115,14 +115,15 @@ def pre_dataframe() -> Iterable[Res[SleepEntry]]:
|
||||||
yield group[0]
|
yield group[0]
|
||||||
else:
|
else:
|
||||||
err = RuntimeError(f'Multiple sleeps per night, not supported yet: {group}')
|
err = RuntimeError(f'Multiple sleeps per night, not supported yet: {group}')
|
||||||
set_error_datetime(err, dt=dd)
|
set_error_datetime(err, dt=dd) # type: ignore[arg-type]
|
||||||
logger.exception(err)
|
logger.exception(err)
|
||||||
yield err
|
yield err
|
||||||
|
|
||||||
|
|
||||||
def dataframe():
|
def dataframe():
|
||||||
dicts: List[Dict] = []
|
dicts: List[Dict[str, Any]] = []
|
||||||
for s in pre_dataframe():
|
for s in pre_dataframe():
|
||||||
|
d: Dict[str, Any]
|
||||||
if isinstance(s, Exception):
|
if isinstance(s, Exception):
|
||||||
dt = extract_error_datetime(s)
|
dt = extract_error_datetime(s)
|
||||||
d = {
|
d = {
|
||||||
|
@ -141,7 +142,7 @@ def dataframe():
|
||||||
}
|
}
|
||||||
dicts.append(d)
|
dicts.append(d)
|
||||||
|
|
||||||
import pandas as pd # type: ignore
|
import pandas as pd
|
||||||
return pd.DataFrame(dicts)
|
return pd.DataFrame(dicts)
|
||||||
# TODO tz is in sleeps json
|
# TODO tz is in sleeps json
|
||||||
|
|
||||||
|
|
|
@ -15,7 +15,7 @@ from typing import Dict, Any, NamedTuple
|
||||||
# print(line)
|
# print(line)
|
||||||
|
|
||||||
import matplotlib.pyplot as plt # type: ignore
|
import matplotlib.pyplot as plt # type: ignore
|
||||||
from numpy import genfromtxt # type: ignore
|
from numpy import genfromtxt
|
||||||
import matplotlib.pylab as pylab # type: ignore
|
import matplotlib.pylab as pylab # type: ignore
|
||||||
|
|
||||||
pylab.rcParams['figure.figsize'] = (32.0, 24.0)
|
pylab.rcParams['figure.figsize'] = (32.0, 24.0)
|
||||||
|
@ -109,7 +109,7 @@ dates = [parse_date(u.date, yearfirst=True, dayfirst=False) for u in useful]
|
||||||
|
|
||||||
# TODO don't need this anymore? it's gonna be in dashboards package
|
# TODO don't need this anymore? it's gonna be in dashboards package
|
||||||
from kython.plotting import plot_timestamped # type: ignore
|
from kython.plotting import plot_timestamped # type: ignore
|
||||||
for attr, lims, mavg, fig in [ # type: ignore
|
for attr, lims, mavg, fig in [
|
||||||
('light', (0, 400), 5, None),
|
('light', (0, 400), 5, None),
|
||||||
('deep', (0, 600), 5, None),
|
('deep', (0, 600), 5, None),
|
||||||
('total', (200, 600), 5, None),
|
('total', (200, 600), 5, None),
|
||||||
|
@ -128,7 +128,7 @@ for attr, lims, mavg, fig in [ # type: ignore
|
||||||
if mavg is not None:
|
if mavg is not None:
|
||||||
mavgs.append((mavg, 'green'))
|
mavgs.append((mavg, 'green'))
|
||||||
fig = plot_timestamped(
|
fig = plot_timestamped(
|
||||||
dts, # type: ignore
|
dts,
|
||||||
[getattr(u, attr) for u in useful],
|
[getattr(u, attr) for u in useful],
|
||||||
marker='.',
|
marker='.',
|
||||||
ratio=(16, 4),
|
ratio=(16, 4),
|
||||||
|
|
Loading…
Add table
Reference in a new issue