mgyigit commited on
Commit
8e737a7
·
verified ·
1 Parent(s): ea30d94

Update src/vis_utils.py

Browse files
Files changed (1) hide show
  1. src/vis_utils.py +2 -1
src/vis_utils.py CHANGED
@@ -5,6 +5,7 @@ import os
5
  import json
6
  import yaml
7
  import matplotlib.pyplot as plt
 
8
  import seaborn as sns
9
  import plotnine as p9
10
  import sys
@@ -170,7 +171,7 @@ def plot_function_results(method_names, aspect, metric, function_path="/tmp/func
170
  return filename
171
 
172
  def plot_family_results(methods_selected, dataset, metric, family_path="/tmp/family_results.csv"):
173
- if not os.path.exists(function_path):
174
  benchmark_types = ["similarity", "function", "family", "affinity"] #download all files for faster results later
175
  download_from_hub(benchmark_types)
176
 
 
5
  import json
6
  import yaml
7
  import matplotlib.pyplot as plt
8
+ from matplotlib import ticker
9
  import seaborn as sns
10
  import plotnine as p9
11
  import sys
 
171
  return filename
172
 
173
  def plot_family_results(methods_selected, dataset, metric, family_path="/tmp/family_results.csv"):
174
+ if not os.path.exists(family_path):
175
  benchmark_types = ["similarity", "function", "family", "affinity"] #download all files for faster results later
176
  download_from_hub(benchmark_types)
177