h2oaimichalmarszalek commited on
Commit
20638d1
·
1 Parent(s): 97ee047

refactor to functor

Browse files
Files changed (1) hide show
  1. tools/helm_test_gen.py +1 -4
tools/helm_test_gen.py CHANGED
@@ -1,6 +1,5 @@
1
  import re
2
  import yaml
3
- import os
4
  from typing import Optional
5
  from smolagents import Tool
6
  from jinja2 import Environment, FileSystemLoader
@@ -67,14 +66,12 @@ class GenerateHelmTest(Tool):
67
  """
68
  doc = [*yaml.safe_load_all(document)]
69
  result = ''
70
- current_dir = os.path.dirname(os.path.abspath(__file__))
71
- template_dir = os.path.join(current_dir, 'templates')
72
 
73
  for item in doc:
74
  name = item['metadata']['name']
75
  kind = item['kind']
76
  kindDesc = (re.sub(r'((?<=[a-z])[A-Z]|(?<!\A)[A-Z](?=[a-z]))', r' \1', kind)).lower()
77
- env = Environment(loader=FileSystemLoader(template_dir))
78
  env.globals['gyv'] = GenerateHelmTest.get_yaml_value
79
  env.globals['bool_to_str'] = GenerateHelmTest.bool_to_str
80
  env.globals['get_object_by_kind'] = GenerateHelmTest.get_object_by_kind
 
1
  import re
2
  import yaml
 
3
  from typing import Optional
4
  from smolagents import Tool
5
  from jinja2 import Environment, FileSystemLoader
 
66
  """
67
  doc = [*yaml.safe_load_all(document)]
68
  result = ''
 
 
69
 
70
  for item in doc:
71
  name = item['metadata']['name']
72
  kind = item['kind']
73
  kindDesc = (re.sub(r'((?<=[a-z])[A-Z]|(?<!\A)[A-Z](?=[a-z]))', r' \1', kind)).lower()
74
+ env = Environment(loader=FileSystemLoader('templates'))
75
  env.globals['gyv'] = GenerateHelmTest.get_yaml_value
76
  env.globals['bool_to_str'] = GenerateHelmTest.bool_to_str
77
  env.globals['get_object_by_kind'] = GenerateHelmTest.get_object_by_kind