Add length check guards for delete methods

BeastyBlacksmith · Jan 2, 2024 · 29be0e5 · 29be0e5
1 parent 019687e
commit 29be0e5
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/ProtoStruct.jl b/src/ProtoStruct.jl
@@ -116,8 +116,12 @@ macro proto(expr)
                         error("The supertype of a proto struct is not redefinable. Please restart your julia session.")
                     end
                     the_methods = collect(methods($name))
-                    Base.delete_method(the_methods[1])
-                    Base.delete_method(the_methods[2])
+                    if length(the_methods) >= 1
+                        Base.delete_method(the_methods[1])
+                    end
+                    if length(the_methods) >= 2
+                        Base.delete_method(the_methods[2])
+                    end
                 end
 
                 function $name($(fields...)) where {$(type_parameters...)}
@@ -181,8 +185,12 @@ macro proto(expr)
                         error("The supertype of a proto struct is not redefinable. Please restart your julia session.")
                     end
                     the_methods = collect(methods($name))
-                    Base.delete_method(the_methods[1])
-                    Base.delete_method(the_methods[2])
+                    if length(the_methods) >= 1
+                        Base.delete_method(the_methods[1])
+                    end
+                    if length(the_methods) >= 2
+                        Base.delete_method(the_methods[2])
+                    end
                 end
 
                 function $name($(fields...)) where {$(type_parameters...)}