1 files changed, 12 insertions, 1 deletions
diff --git a/gradle/src/main/groovy/com/android/build/gradle/internal/model/VariantImpl.java b/gradle/src/main/groovy/com/android/build/gradle/internal/model/VariantImpl.java
index f9a5c86..2baef84 100644
--- a/gradle/src/main/groovy/com/android/build/gradle/internal/model/VariantImpl.java
+++ b/gradle/src/main/groovy/com/android/build/gradle/internal/model/VariantImpl.java
@@ -19,6 +19,7 @@ package com.android.build.gradle.internal.model;
 import com.android.annotations.NonNull;
 import com.android.annotations.Nullable;
 import com.android.builder.model.ArtifactInfo;
+import com.android.builder.model.SourceProvider;
 import com.android.builder.model.Variant;
 import com.android.builder.model.ProductFlavor;
 
@@ -46,6 +47,8 @@ class VariantImpl implements Variant, Serializable {
     private final ArtifactInfo mainArtifactInfo;
     @Nullable
     private final ArtifactInfo testArtifactInfo;
+    @Nullable
+    private final SourceProvider variantSourceProvider;
 
     VariantImpl(@NonNull  String name,
                 @NonNull  String displayName,
@@ -53,7 +56,8 @@ class VariantImpl implements Variant, Serializable {
                 @NonNull  List<String> productFlavorNames,
                 @NonNull  ProductFlavorImpl mergedFlavor,
                 @NonNull  ArtifactInfo mainArtifactInfo,
-                @Nullable ArtifactInfo testArtifactInfo) {
+                @Nullable ArtifactInfo testArtifactInfo,
+                @Nullable SourceProvider variantSourceProvider) {
         this.name = name;
         this.displayName = displayName;
         this.buildTypeName = buildTypeName;
@@ -61,6 +65,7 @@ class VariantImpl implements Variant, Serializable {
         this.mergedFlavor = mergedFlavor;
         this.mainArtifactInfo = mainArtifactInfo;
         this.testArtifactInfo = testArtifactInfo;
+        this.variantSourceProvider = variantSourceProvider;
     }
 
     @Override
@@ -110,4 +115,10 @@ class VariantImpl implements Variant, Serializable {
     public List<String> getResourceConfigurations() {
         return Collections.emptyList();
     }
+
+    @Override
+    @Nullable
+    public SourceProvider getSourceProvider() {
+        return variantSourceProvider;
+    }
 }