|  | @@ -28,12 +28,12 @@ def _get_staged_proto_file(context, source_file):
 | 
	
		
			
				|  |  |  def _generate_py_impl(context):
 | 
	
		
			
				|  |  |      protos = []
 | 
	
		
			
				|  |  |      for src in context.attr.deps:
 | 
	
		
			
				|  |  | -        for file in src.proto.direct_sources:
 | 
	
		
			
				|  |  | +        for file in src[ProtoInfo].direct_sources:
 | 
	
		
			
				|  |  |              protos.append(_get_staged_proto_file(context, file))
 | 
	
		
			
				|  |  |      includes = [
 | 
	
		
			
				|  |  |          file
 | 
	
		
			
				|  |  |          for src in context.attr.deps
 | 
	
		
			
				|  |  | -        for file in src.proto.transitive_imports.to_list()
 | 
	
		
			
				|  |  | +        for file in src[ProtoInfo].transitive_imports.to_list()
 | 
	
		
			
				|  |  |      ]
 | 
	
		
			
				|  |  |      proto_root = get_proto_root(context.label.workspace_root)
 | 
	
		
			
				|  |  |      format_str = (_GENERATED_GRPC_PROTO_FORMAT if context.executable.plugin else _GENERATED_PROTO_FORMAT)
 |