mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-08 15:28:26 +01:00
Paweł Redzyński
9940540917
* metrics: include revision information when collecting metrics files * plots: test warning message on no plot on revision * plots/metrics: move to one package, introduce common tests * tests: fix utils placement
24 lines
627 B
Python
24 lines
627 B
Python
import csv
|
|
import json
|
|
|
|
from funcy import first
|
|
|
|
|
|
def _write_csv(metric, filename, header=True):
|
|
with open(filename, "w", newline="") as csvobj:
|
|
if header:
|
|
writer = csv.DictWriter(
|
|
csvobj, fieldnames=list(first(metric).keys())
|
|
)
|
|
writer.writeheader()
|
|
writer.writerows(metric)
|
|
else:
|
|
writer = csv.writer(csvobj)
|
|
for d in metric:
|
|
assert len(d) == 1
|
|
writer.writerow(list(d.values()))
|
|
|
|
|
|
def _write_json(tmp_dir, metric, filename):
|
|
tmp_dir.gen(filename, json.dumps(metric, sort_keys=True))
|