Merge pull request citusdata#177 from mpantel/mpantel-patch-Rails-7.1-compatibility

Update model_extensions.rb

Author: serprex

lib/activerecord-multi-tenant/model_extensions.rb

@@ -30,7 +30,7 @@ def partition_key
 
           # Avoid primary_key errors when using composite primary keys (e.g. id, tenant_id)
           def primary_key
-            return @primary_key if @primary_key
+            return @primary_key if defined?(PRIMARY_KEY_NOT_SET) ? !PRIMARY_KEY_NOT_SET.equal?(@primary_key) : @primary_key
 
             primary_object_keys = Array.wrap(connection.schema_cache.primary_keys(table_name)) - [partition_key]