Commit b8110fe

committed

Fix n sample size plotting. Add type hint to plotting methods

1 parent 6319dae commit b8110feCopy full SHA for b8110fe

File tree

-8

lines changed

-8

lines changed

Lines changed: 8 additions & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -96,17 +96,17 @@ def plot_results(`
`96`	`96`	`def plot_residues_correlation(`
`97`	`97`	`data=None,`
`98`	`98`	`*,`
`99`		`- figsize=(10, 6),`
`100`		`- n=100,`
`101`		`- style="default",`
`102`		`- facecolor="white",`
`103`		`- title="Residual Analysis",`
`104`		`- ylabel="Correlation",`
`105`		`-):`
	`99`	`+ figsize: Tuple[int, int] =(10, 6),`
	`100`	`+ n: int=100,`
	`101`	`+ style: str="default",`
	`102`	`+ facecolor: str="white",`
	`103`	`+ title: str="Residual Analysis",`
	`104`	`+ ylabel: str="Correlation",`
	`105`	`+)->None:`
`106`	`106`	`plt.style.use(style)`
`107`	`107`	`plt.rcParams["axes.facecolor"] = facecolor`
`108`	`108`	`_, ax = plt.subplots(figsize=figsize, facecolor=facecolor)`
`109`		`- ax.plot(data[0], color="#1f77b4")`
	`109`	`+ ax.plot(data[0][:n], color="#1f77b4")`
`110`	`110`	`ax.axhspan(data[1], data[2], color="#ccd9ff", alpha=0.5, lw=0)`
`111`	`111`	`ax.set_xlabel("Lag", fontsize=14)`
`112`	`112`	`ax.set_ylabel(ylabel, fontsize=14)`

Comments

(0)