lib/ridley/chef_object.rb in ridley-1.1.1 vs lib/ridley/chef_object.rb in ridley-1.2.0

@@ -1,14 +1,16 @@
+require 'varia_model'
+
 module Ridley
   class ChefObject
     class << self
       # @return [String, nil]
       def chef_id
         @chef_id
       end
 
-      # @param [String, Symbol] identifier
+      # @param [#to_sym] identifier
       #
       # @return [String]
       def set_chef_id(identifier)
         @chef_id = identifier.to_sym
       end
@@ -16,11 +18,11 @@
       # @return [String]
       def chef_type
         @chef_type ||= self.class.name.underscore
       end
 
-      # @param [String, Symbol] type
+      # @param [#to_s] type
       #
       # @return [String]
       def set_chef_type(type)
         @chef_type = type.to_s
         attribute(:chef_type, default: type)
@@ -38,10 +40,10 @@
         @chef_json_class = klass
         attribute(:json_class, default: klass)
       end
     end
 
-    include Chozo::VariaModel
+    include VariaModel
     include Comparable
 
     # @param [Ridley::Resource] resource
     # @param [Hash] new_attrs
     def initialize(resource, new_attrs = {})