Merge pull request #12 from w-e-w/fix-xyz-for-1.9

Fix xyz for 1.9
v0xie · Apr 22, 2024 · 8c2fd24 · 8c2fd24
2 parents 124355d + 8936823
commit 8c2fd24
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+__pycache__/
diff --git a/scripts/agat.py b/scripts/agat.py
@@ -249,7 +249,7 @@ def fun(p, x, xs):
     return fun
 
 def make_axis_options():
-        xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module
+        xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ in ('scripts.xyz_grid', 'xyz_grid.py')][0].module
         extra_axis_options = {
                 xyz_grid.AxisOption("[AgentAttention] Active", str, aa_apply_override('aa_active', boolean=True), choices=xyz_grid.boolean_choice(reverse=True)),
                 xyz_grid.AxisOption("[AgentAttention] Use Second Pass", str, aa_apply_override('aa_use_sp', boolean=True), choices=xyz_grid.boolean_choice(reverse=True)),